hihoCoder#1055 刷油漆

原题地址

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

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

代码：

#include <iostream>
#include <cstring>

using namespace std;

#define SIZE 128

int tree[SIZE][SIZE];
int v[SIZE];
int f[SIZE][SIZE];
int N, M;

void traverse(int t, int p) {
  f[t][0] = 0;
  f[t][1] = v[t];

  for (int i = 1; i <= tree[t][0]; i++) {
    int t_child = tree[t][i];
    if (t_child == p)
      continue;
    traverse(t_child, t);
    for (int m = M; m >= 2; m--) {
      for (int m_child = 1; m_child <= m - 1; m_child++) {
        f[t][m] = max(f[t][m], f[t][m - m_child] + f[t_child][m_child]);
      }
    }
  }
}

int main() {
  int res = 0;

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

  cin >> N >> M;
  for (int i = 1; i <= N; i++)
    cin >> v[i];
  for (int i = 1; i <= N - 1; i++) {
    int a, b;
    cin >> a >> b;
    tree[a][0]++;
    tree[b][0]++;
    tree[a][tree[a][0]] = b;
    tree[b][tree[b][0]] = a;
  }

  traverse(1, -1);
  cout << f[1][M] << endl;

  return 0;
}