结点选择

问题描述

有一棵 n 个节点的树，树上每个节点都有一个正整数权值。如果一个点被选择了，那么在树上和它相邻的点都不能被选择。求选出的点的权值和最大是多少？

输入格式

第一行包含一个整数 n 。

接下来的一行包含 n 个正整数，第 i 个正整数代表点 i 的权值。

接下来一共 n-1 行，每行描述树上的一条边。

输出格式

输出一个整数，代表选出的点的权值和的最大值。

样例输入

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

样例输出

12

样例说明

选择3、4、5号点，权值和为 3+4+5 = 12 。

数据规模与约定

对于20%的数据， n <= 20。

对于50%的数据， n <= 1000。

对于100%的数据， n <= 100000。

权值均为不超过1000的正整数。

一开始没能完全理解题意,以为它的数据是顺序给出的,结果一个数据也没对,之后看了网上的几个答案,发现我把题目想太简单了,网上的答案都是递归求解的,跟我的很不同.不想放弃自己的思路,结果几个小时就这么过去了.不过最终还是坚持着改完了.

1. #include<stdio.h>
2. #include<stdlib.h>
3. #include<string.h>
5. typedef struct Node
6. {
7.     int to;
8.     int next;
9. }Node;
11. #define N 100020
13. int max(int a, int b)
14. {
15.     return a > b ? a : b;
16. }
18. int on[N], off[N];
20. int rel[N];
21. Node relBus[2 * N];
22. int relBusTop = 1;
24. int queue[N] = {1};
25. int qStart = 0, qEnd = 1;
27. int checked[N] = {0, 1};
29. int ser[N];
30. int sp = 0;
32. int main(void)
33. {
34.     int n, i, j;
35.     
36.     scanf("%d", &n);
37.     for(i = 1; i <= n; i++)
38.     {
39.         scanf("%d", &on[i]);
40.         off[i] = 0;
41.     }
42. 
    
43.     for(i = 0; i < n - 1; i++)
44.     {
45.         int a, b;
46.         scanf("%d%d", &a, &b);
47.         
48.         relBus[relBusTop].to = b;
49.         relBus[relBusTop].next = rel[a];
50.         rel[a] = relBusTop++;
51.         
52.         relBus[relBusTop].to = a;
53.         relBus[relBusTop].next = rel[b];
54.         rel[b] = relBusTop++;
55.     }
57.     while(qStart < qEnd)
58.     {
59.         int now = queue[qStart++];
60.         ser[sp++] = now;
61.         int p = rel[now];
62.         while(p > 0)
63.         {
64.             int son = relBus[p].to;
65.             if(checked[son] == 0)
66.             {
67.                 queue[qEnd++] = son;
68.                 checked[son] = 1;
69.             }
70.             p = relBus[p].next;
71.         }
72.     }
73.     
74.     for(i = n - 1; i >= 0; i--)
75.     {
76.         int son = ser[i];
77.         int p = rel[son];
78.         
79.         while(p > 0)
80.         {
81.             int father = relBus[p].to;
82.             
83.             on[father] += off[son];
84.             off[father] += max(on[son], off[son]);
85.             p = relBus[p].next;
86.         }
87.     }
88.     
89.     printf("%d", max(on[1], off[1]));
90.     return 0;
91. }