标签:超过 树形dp main clu i++ NPU clear color des
它们之间由一些小路相连,小路不会形成环,即两个宝藏点之间有且只有一条通路
那么我们不知道在f[i][j-1-k]中是否有v[i],如果有就算重了
可以多定一个数组g,也可以多加一维,其实是一样的
g[i][j]表示 ]以i为根的子树,剩j,不挖i 的的最大价值
1 #include<cstdio> 2 #include<iostream> 3 #include<algorithm> 4 using namespace std; 5 int n,m,cnt,v[110],f[110][110],to[210],from[210],head[110],k[110],g[110][110]; 6 void insert(int x,int y) { to[++cnt]=y; from[cnt]=head[x]; head[x]=cnt; } 7 void dp(int root,int fa) 8 { 9 f[root][1]=v[root]; 10 for (int w=head[root];w;w=from[w]) 11 if (to[w]!=fa) 12 { 13 dp(to[w],root); 14 for (int i=1;i<=m;i++) k[i]=max(f[root][i],f[to[w]][i]); 15 for (int i=1;i<=m;i++) 16 for (int j=1;j<i;j++) 17 k[i]=max(k[i],g[root][i-j-1]+f[to[w]][j]+v[root]); 18 for (int i=1;i<=m;i++) f[root][i]=k[i]; 19 for (int i=1;i<=m;i++) k[i]=max(f[to[w]][i],g[root][i]); 20 for (int i=1;i<=m;i++) 21 for (int j=1;j<i;j++) 22 k[i]=max(k[i],g[root][i-j]+f[to[w]][j]); 23 for (int i=1;i<=m;i++) g[root][i]=k[i]; 24 } 25 } 26 int main() 27 { 28 scanf("%d%d",&n,&m); 29 for (int i=1;i<=n;i++) scanf("%d",&v[i]); 30 for (int i=1;i<=n-1;i++) 31 { 32 int x,y; 33 scanf("%d%d",&x,&y); 34 insert(x,y); insert(y,x); 35 } 36 dp(1,0); 37 printf("%d",f[1][m]); 38 return 0; 39 }
标签:超过 树形dp main clu i++ NPU clear color des
原文地址:https://www.cnblogs.com/Comfortable/p/9277479.html