hihoCoder#1055 刷油漆

原题地址

 

第一次做树的动归题，如果没有提示的话还是挺难的

提示里的递推式隐含了状态压缩（m从大往小遍历），不是那么好想，只能说不能再屌了。

 

代码：

 1 #include <iostream>

 2 #include <cstring>

 3 

 4 using namespace std;

 5 

 6 #define SIZE 128

 7 

 8 int tree[SIZE][SIZE];

 9 int v[SIZE];

10 int f[SIZE][SIZE];

11 int N, M;

12 

13 void traverse(int t, int p) {

14   f[t][0] = 0;

15   f[t][1] = v[t];

16 

17   for (int i = 1; i <= tree[t][0]; i++) {

18     int t_child = tree[t][i];

19     if (t_child == p)

20       continue;

21     traverse(t_child, t);

22     for (int m = M; m >= 2; m--) {

23       for (int m_child = 1; m_child <= m - 1; m_child++) {

24         f[t][m] = max(f[t][m], f[t][m - m_child] + f[t_child][m_child]);

25       }

26     }

27   }

28 }

29 

30 int main() {

31   int res = 0;

32 

33   memset(tree, 0, SIZE * SIZE * sizeof(int));

34   memset(f, 0, SIZE * SIZE * sizeof(int));

35 

36   cin >> N >> M;

37   for (int i = 1; i <= N; i++)

38     cin >> v[i];

39   for (int i = 1; i <= N - 1; i++) {

40     int a, b;

41     cin >> a >> b;

42     tree[a][0]++;

43     tree[b][0]++;

44     tree[a][tree[a][0]] = b;

45     tree[b][tree[b][0]] = a;

46   }

47 

48   traverse(1, -1);

49   cout << f[1][M] << endl;

50 

51   return 0;

52 }