标签:tin from bsp output name 代码 需求量 code space
1 #include <cstdio> 2 #include <iostream> 3 #include <cmath> 4 #include <cstring> 5 using namespace std; 6 struct edge { int to,from; }e[510*2]; 7 int f[510][510][2],n,m,a[510],head[510],cnt; 8 void insert(int x,int y) { e[++cnt].to=y; e[cnt].from=head[x]; head[x]=cnt; } 9 void dp(int x,int fa) 10 { 11 for (int i=head[x];i;i=e[i].from) 12 { 13 if (e[i].to==fa) continue; 14 dp(e[i].to,x); 15 for (int j=m;j>=1;j--) 16 for (int k=0;k<=m;k++) 17 { 18 if (j-k>=2) 19 f[x][j][1]=max(f[x][j][1],f[x][j-k-2][1]+f[e[i].to][k][0]), 20 f[x][j][0]=max(f[x][j][0],f[x][j-k-2][0]+f[e[i].to][k][0]); 21 if (j-k>=1) f[x][j][1]=max(f[x][j][1],f[x][j-k-1][0]+f[e[i].to][k][1]); 22 } 23 } 24 for (int i=m;i>=1;i--) 25 { 26 f[x][i][0]=max(f[x][i][0],f[x][i-1][0]+a[x]); 27 f[x][i][1]=max(f[x][i][1],f[x][i-1][1]+a[x]); 28 } 29 } 30 int main() 31 { 32 freopen("dostavljac.in","r",stdin); 33 freopen("dostavljac.out","w",stdout); 34 scanf("%d%d",&n,&m); 35 for (int i=1;i<=n;i++) scanf("%d",&a[i]); 36 for (int i=1;i<=n-1;i++) 37 { 38 int u,v; 39 scanf("%d%d",&u,&v); 40 insert(u,v); insert(v,u); 41 } 42 dp(1,0); 43 printf("%d",f[1][m][1]); 44 return 0; 45 }
标签:tin from bsp output name 代码 需求量 code space
原文地址:https://www.cnblogs.com/Comfortable/p/9451287.html