POJ 1155 TELE(树形DP)

来源：互联网发布：菠菜代源码论坛编辑：程序博客网时间：2024/06/05 17:23

第一道没看题解，自己写出来的树形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 0x3f3f3f3fint 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;}

0 0