bzoj1722 [Usaco2006 Mar] Milk Team Select 产奶比赛（树形dp）

dp[i][j][0/1]表示以i为根的子树,有j对关系，根选没选的最大产奶量。则
dp[i][j][0]=max{tmp[j-k][0]+dp[y][k][0/1]|0<=k<=j}
dp[i][j][1]=max{tmp[j-k][1]+dp[y][k][0]/dp[y][k-1][1]|0<=k<=j}+a[i]
复杂度O(n^3)，注意一些细节，过了应该没问题。转移很像背包dp的转移，也可以说是很像左儿子右兄弟的想法。注意j倒着推即可。

#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define N 510
#define inf 0x3f3f3f3f
inline int read(){
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
    return x*f;
}
int n,kk,fa[N],a[N],sz[N],dp[N][2][N];//dp[i][0][j]--以i为根的子树,有j对关系，根选没选的最大产奶量
vector<int>son[N];
void dfs(int x){
    sz[x]=1;int tmp[2][N];for(int i=0;i<=n;++i) tmp[0][i]=tmp[1][i]=-inf;tmp[0][0]=tmp[1][0]=0;
    for(int i=0;i<son[x].size();++i){
        int y=son[x][i];dfs(y);sz[x]+=sz[y];
        for(int j=sz[x]-1;j>=0;--j){
            for(int k=0;k<=j&&k<sz[y];++k) tmp[0][j]=max(tmp[0][j],tmp[0][j-k]+max(dp[y][0][k],dp[y][1][k]));
            if(x==0) continue;
            for(int k=0;k<=j&&k<sz[y];++k) tmp[1][j]=max(tmp[1][j],tmp[1][j-k]+dp[y][0][k]);
            for(int k=0;k<j&&k<sz[y];++k) tmp[1][j]=max(tmp[1][j],tmp[1][j-k-1]+dp[y][1][k]);
        }
    }if(x!=0) for(int j=0;j<sz[x];++j) tmp[1][j]+=a[x];memcpy(dp[x],tmp,sizeof(tmp));
}
int main(){
//  freopen("a.in","r",stdin);
    n=read();kk=read();
    for(int i=1;i<=n;++i)
        for(int j=1;j<=n;++j) dp[i][0][j]=dp[i][1][j]=-inf;
    for(int i=1;i<=n;++i) a[i]=read(),fa[i]=read(),son[fa[i]].push_back(i);
    dfs(0);int ans=-1;
    for(int j=sz[0]-1;j>=0;--j) if(dp[0][0][j]>=kk){ans=j;break;}
    printf("%d\n",ans);
    return 0;
}