P1364 医院设置

题目描述

设有一棵二叉树，如图：

其中，圈中的数字表示结点中居民的人口。圈边上数字表示结点编号，现在要求在某个结点上建立一个医院，使所有居民所走的路程之和为最小，同时约定，相邻接点之间的距离为 11。如上图中，若医院建在 1 处，则距离和 =4+12+2×20+2×40=136；若医院建在 3 处，则距离和 =4×2+13+20+40=81。

输入格式

第一行一个整数 𝑛，表示树的结点数。

接下来的 𝑛 行每行描述了一个结点的状况，包含三个整数 𝑤,𝑢,𝑣，其中 𝑤 为居民人口数，𝑢 为左链接（为 0 表示无链接），𝑣 为右链接（为 0 表示无链接）。

输出格式

一个整数，表示最小距离和。

代码:

package lanqiao;

import java.util.*;
import java.util.Scanner;

class TreeNode {
    int val;
    int left;
    int right;
    int father;

}

public class Main{

    static int n, sum, ans = Integer.MAX_VALUE;
    static TreeNode[] t = new TreeNode[106];
    static int[] vis = new int[105];
    //第i个结点，深度
    public static int dfs(int step, int pos) {
        if (vis[step] == 1 || step == 0) {
            return 0;
        }
        vis[step] = 1;

        int fa = t[step].father, l = t[step].left, r = t[step].right;

        int x = dfs(l, pos + 1) + dfs(r, pos + 1) + dfs(fa, pos + 1) + t[step].val * pos;
        vis[step] = 0;
        return x;
    }

    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        n = sc.nextInt();

        for (int i = 1; i <= n; i++) {
            t[i] = new TreeNode();
            t[i].val = sc.nextInt();
            t[i].left = sc.nextInt();
            t[i].right = sc.nextInt();
            vis[i] = 0;
        }
        t[1].father = 0;

        for (int i = 1; i <= n; i++) {

            if (t[i].left != 0)
                t[t[i].left].father = i;//左节点的父节点
            if (t[i].right != 0)
                t[t[i].right].father = i;//右节点的父节点
        }

        for (int i = 1; i <= n; i++) {
            if (vis[i] != 1) {//未被访问过

                ans = Math.min(ans, dfs(i, 0));
            }
        }
        System.out.println(ans);
    }
}