蓝桥杯-生命之树

在X森林里，上帝创建了生命之树。
他给每棵树的每个节点（叶子也称为一个节点）上，都标了一个整数，代表这个点的和谐值。
上帝要在这棵树内选出一个非空节点集S，使得对于S中的任意两个点a,b，都存在一个点列 {a, v1, v2, …, vk, b} 使得这个点列中的每个点都是S里面的元素，且序列中相邻两个点间有一条边相连。
在这个前提下，上帝要使得S中的点所对应的整数的和尽量大。
这个最大的和就是上帝给生命之树的评分。

经过atm的努力，他已经知道了上帝给每棵树上每个节点上的整数。但是由于 atm 不擅长计算，他不知道怎样有效的求评分。他需要你为他写一个程序来计算一棵树的分数。

「输入格式」
第一行一个整数 n 表示这棵树有 n 个节点。
第二行 n 个整数，依次表示每个节点的评分。
接下来 n-1 行，每行 2 个整数 u, v，表示存在一条 u 到 v 的边。由于这是一棵树，所以是不存在环的。

「输出格式」
输出一行一个数，表示上帝给这棵树的分数。

「样例输入」
5
1 -2 -3 4 5
4 2
3 1
1 2
2 5

「样例输出」
8

「数据范围」
对于 30% 的数据，n <= 10
对于 100% 的数据，0 < n <= 10^5, 每个节点的评分的绝对值不超过 10^6 。

资源约定：
峰值内存消耗 < 256M
CPU消耗 < 3000ms

import java.util.*;
public class Main {
	public static int ans;
	public static int n;
	public static int[] v;
	public static int[][] mapp;
	public static int[] sum;
    public static void main(String []args) {
		Scanner scanner = new Scanner(System.in);
		n = scanner.nextInt();
	    v = new int[n+10];
		for(int i=1;i<=n;i++){
			v[i] = scanner.nextInt();
		}
		mapp = new int[n+10][n+10];
		for(int i=1;i<=n-1;i++){
			int a = scanner.nextInt();
			int b = scanner.nextInt();
			mapp[a][b] = 1;
			mapp[b][a] = 1;
				
		}
	    sum = new int[n+10];
        dfs(1,0);
		System.out.println(ans);
    }
		public static void dfs(int root,int father){
		sum[root] = v[root];
		for(int i=1;i<=n;i++){
			if(mapp[root][i]==1&&i!=father){
				int son = i;
				dfs(i,root);
				if(sum[i]>0){
					sum[root] += sum[i];
				}
			}
		}
		ans = Math.max(ans,sum[root]);
	}
}

笔记：
1.无根树转有根树模板

void dfs(int root,int father){
	for(int i=0;i<vec.size();i++){
		int son = vec[root][i];
		if(son!=father){
			dfs(son,root);
		}
	}
}

Java用list代替vector写邻接表，写法如下:

import java.util.*;
public class Main {
	public static long ans;
	public static int n;
	public static long[] v;
	public static int[][] mapp;
	public static long[] sum;
	public static List<Integer>[] list;
    public static void main(String []args) {
		Scanner scanner = new Scanner(System.in);
		n = scanner.nextInt();
	    v = new long[n+10];
		for(int i=1;i<=n;i++){
			v[i] = scanner.nextInt();
		}
		list = new ArrayList[1000];
		for(int i=0;i<1000;i++){
			list[i] = new ArrayList<Integer>();
		}
		for(int i=1;i<=n-1;i++){
			int a = scanner.nextInt();
			int b = scanner.nextInt();
			list[a].add(b);
			list[b].add(a);
				
		}
	    sum = new long[n+10];
        dfs(1,0);
		System.out.println(ans);
    }
		public static void dfs(int root,int father){
		sum[root] = v[root];
		for(int i=0;i<list[root].size();i++){
			if(list[root].get(i)!=father){
				Integer son = list[root].get(i);
				dfs(son,root);
				if(sum[son]>0){
					sum[root] += sum[son];
				}
			}
		}
		ans = Math.max(ans,sum[root]);
	}
}

笔记：

1.java可以用ArrayList可变长数组代替vector，做邻接表。定义一维ArrayList用，List<Integer>[] list = new ArrayList[maxn+5];注意在main里需要再次将每个list对象进行初始化，因为这个定义只初始化了数组,用
for(int i=0;i<maxn+5;i++){
	list[i] = new ArrrayList<Integer>();
}
对象数组都要单独初始化每个对象

取用元素，第a条list的第i个元素，
for(int i=0;i<list[a].size();i++){
		Integer son = list[a].get(i);//存放Integer对象元素
}
放入元素,list[a].add(b)