ZOJ 3201 Tree of Tree 大水树形背包

题意与做法：题目也不长，就是原始的树形背包，不过粗心了，敲代码的时候卡了一下。。

#include<stdio.h>#include<string.h>#define LMT 105int get[LMT][LMT],dp[LMT][LMT],wei[LMT],next[LMT];typedef struct{    int u,v,next;}line;line e[LMT<<1];int n,st,all;int max(int a,int b){    return a>b?a:b;}void insert(int u,int v){    e[all].u=u;    e[all].v=v;    e[all].next=next[u];    next[u]=all++;    e[all].u=v;    e[all].v=u;    e[all].next=next[v];    next[v]=all++;}void dfs(int u,int pre){    int v,x,j,k;    for(x=next[u];x!=-1;x=e[x].next)    if(e[x].v!=pre)    {        v=e[x].v;        dfs(v,u);        for(j=st;j>=0;j--)          for(k=0;k<=j;k++)          if(get[u][j-k]!=-1)          get[u][j]=max(get[u][j],get[u][j-k]+dp[v][k]);    }    dp[u][1]=wei[u];    for(j=st;j>=1;j--)     for(k=0;k<=j-1;k++)//k竟然写成了j。。     if(dp[u][j-k]!=-1)     dp[u][j]=max(dp[u][j],dp[u][j-k]+get[u][k]);}int main(){    int i;    while(scanf("%d%d",&n,&st)!=EOF)    {        for(i=0;i<n;i++)        scanf("%d",&wei[i]);        memset(dp,-1,sizeof(dp));        memset(get,-1,sizeof(get));        memset(next,-1,sizeof(next));        all=0;        for(i=0;i<n;i++)            get[i][0]=dp[i][0]=0;        for(i=1;i<n;i++)        {            int u,v;            scanf("%d%d",&u,&v);            insert(u,v);        }        dfs(0,0);        int ans=0;//忘记赋初值了。。。        for(i=0;i<n;i++)        ans=max(ans,dp[i][st]);        printf("%d\n",ans);    }    return 0;}