POJ 1155 TELE(树形DP)

第一道没看题解，自己写出来的树形DP,纪念一下。，

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxn = 3000+5;
int head[maxn],pay[maxn],n,m;
bool vis[maxn];
#define INF 0x3f3f3f3f
int pnt[2*maxn],nxt[2*maxn],cst[2*maxn],cnt;
int dp[maxn][maxn],num[maxn];
void init() {
    memset(head,-1,sizeof(head));
    memset(nxt,-1,sizeof(nxt));
    cnt=0;
}
void add_edge(int a,int b,int w) {
    pnt[cnt]=b;
    cst[cnt]=w;
    nxt[cnt]=head[a];
    head[a]=cnt++;
}
void dfs(int rt) {
    vis[rt]=1;
    dp[rt][0]=0;
    for(int i=1;i<=m;++i) dp[rt][i]=-INF;
    if(rt>n-m){
         dp[rt][1]=pay[rt];
         num[rt]=1;
         return ;
    }
    for(int i=head[rt]; i+1 ; i=nxt[i]) {
        int ch=pnt[i];
        if(!vis[ch]) {
            dfs(ch);
            num[rt]+=num[ch];
            for(int j=m; j>=0; j--)
                for(int k=1; k<=num[ch]&&k+j<=num[rt]; k++)///一开始没加限制条件，直接TLE了
                   dp[rt][j+k]=max(dp[rt][j+k],dp[rt][j]+dp[ch][k]-cst[i]);
        }
    }
}
int main() {
    while(scanf("%d%d",&n,&m)!=EOF) {
        init();
        for(int i=1,k,x,y; i<=n-m; ++i) {
            scanf("%d",&k);
            while(k--) {
                scanf("%d%d",&x,&y);
                add_edge(i,x,y);
            }
        }
        memset(pay,0,sizeof(pay));
        for(int i=n-m+1; i<=n; ++i) scanf("%d",&pay[i]);
        memset(vis,false,sizeof(vis));
        memset(num,0,sizeof(num));
        dfs(1);
        int ans=0;
        for(int i=0; i<=m; ++i)
            if(dp[1][i]>=0) ans=max(ans,i);
        printf("%d\n",ans);
    }
    return 0;
}