算法设计与分析: 4-16 信号增强装置问题

4-16 信号增强装置问题

---

问题描述

各种资源传输网络的功能是将始发地的资源通过网络传输到一个或多个目的地。例如， 通过石油或者天然气输送管网可以将从油田开采的石油和天然气传送给消费者。 同样，通 过高压传输网络可以将发电厂生产的电力传送给用电消费者。为了使问题更具一般性，用术 语信号统称网络中传输的资源 (石油，天然气，电力等等)。各种资源传输网络统称为信号 传输网络。信号经信号传输网络传输时，需要消耗一定的能量，并导致传输能量的衰减(油 压，气压，电压等等)。当传输能量衰减量(压降)达到某个阈值时，将导致传输故障。为 了保证传输畅通，必需在传输网络的适当位置放置信号增强装置，确保传输能量的衰减量不 超过其衰减量容许值。
为了简化问题，假定给定的信号传输网络是以信号始发地为根的一棵树 T。在树 T 的每 一个结点处(除根结点外)可以放置一个信号增强装置。树 T 的结点也代表传输网络的消 费结点。信号经过树 T 的结点传输到其儿子结点。树的每一边上的正权是流经该边的信号 所发生的信号衰减量。信号衰减量是可加的。
信号增强装置问题要求对于一个给定的信号传输网络，计算如何放置最少的信号增强装 置来保证网络传输的畅通。

对于给定的带权树，编程计算放置信号增强装置最少数量。

第一行有 1 个正整数 n，表示给定的带权树有 n 个顶点，编号为 1，2，…，n。编号为 1 的顶点是树根。接下来的 n 行中，第 i+1 行描述与 i 个顶点 相关联的边的信息。每行的第一个正整数 k 表示与该顶点相关联的边数。其后 2k 个数中， 每 2 个数表示 1 条边。第一个数是与该顶点相关联的另一个顶点的编号，第二个数是边权值。 文件的最后一行是正整数 d，表示衰减量容许值。

---

Java

import java.util.Scanner;

public class XinHaoZengqiangZhuangZhi {

    private static int n,p,len,d;
    private static int[] deg,parent,parlen,leaf;
    private static int[] dist;
    private static int[] signal;
    private static boolean[][] existed;

    private static int duplicate;

    public static void main(String[] args){

        Scanner input = new Scanner(System.in);

        while (true){
            n = input.nextInt();

            deg = new int[n+1];
            parent = new int[n+1];
            leaf = new int[n+1];
            parlen = new int[n+1];
            signal = new int[n+1];
            dist = new int[n+1];
            existed = new boolean[n+1][n+1];

            for(int i=1; i<=n; i++){
                deg[i] = input.nextInt();
                duplicate = 0;
                for(int j=0; j<deg[i]; j++){
                    p = input.nextInt();
                    len = input.nextInt();
                    if(i != 1 && (existed[i][p]||existed[p][i])){
                        duplicate++;
                        continue;
                    }
                    parent[p] = i;
                    parlen[p] = len;
                    existed[i][p] = true;
                }
                deg[i] -= duplicate;
                if(deg[i] == 0)
                    leaf[++leaf[0]] = i;
            }
            d = input.nextInt();

            for(int i=1; i<=n; i++)
                if(parlen[i] > d){
                    System.out.println("No solution!");
                    return;
                }

            int result = count();

            System.out.println(result);
        }
    }

    private static int count(){
        int total = 0;
        for(int i=1; i<=leaf[0]; i++)
            if(leaf[i] != 1){//不是根结点
                int plen = parlen[leaf[i]], par = parent[leaf[i]];
                if(signal[leaf[i]]<1 && dist[leaf[i]]+plen>d){//父结点（与之连接的上一级结点）未被删，且该结点当前最大路长（权值）加上到父结点距离（权值）大于d
                    total++;
                    signal[leaf[i]] = 1;
                    par = parent[par];//指向该被减结点的父结点，以便父结点关联边数减去1
                }else if(signal[leaf[i]]<1 && dist[par]<dist[leaf[i]]+plen)//更新结点路长（权值）
                    dist[par] = dist[leaf[i]] + plen;
                if(--deg[par] == 0)//更新结点关联边数
                    leaf[++leaf[0]] = par;//关联边数等于0时 添加叶结点
            }

        return total;
    }
}

---

Input & Output

4
2 2 3 3 1
2 1 3 4 2
1 1 1
1 2 2
4
1

---

Reference

王晓东《计算机算法设计与分析》（第3版）P133-134