LUOGU P1273 有线电视网

传送门

解题思路

明显的树形背包，dp[x][j] 表示以x为根的子树选j个用户的最大值，最后答案取一个最大的j使得dp[1][j] > 0 就行了。

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>

using namespace std;
const int MAXN = 3005;

inline int rd(){
    int x=0,f=1;char ch=getchar();
    while(!isdigit(ch)) {f=ch=='-'?0:1;ch=getchar();}
    while(isdigit(ch))  {x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
    return f?x:-x;
}

int n,m,head[MAXN],cnt,to[MAXN<<1],nxt[MAXN<<1],val[MAXN<<1];
int dp[MAXN][MAXN],ans,w[MAXN],siz[MAXN];

inline void add(int bg,int ed,int ww){
    to[++cnt]=ed,nxt[cnt]=head[bg],val[cnt]=ww,head[bg]=cnt;
}

void dfs(int x,int pre){
    if(x>n-m) {dp[x][1]=w[x],dp[x][0]=0;siz[x]=1;return;}
    for(register int i=head[x];i;i=nxt[i]){
        int u=to[i];if(u==pre) continue;
        dfs(u,x);
        siz[x]+=siz[u];
    }
    dp[x][0]=0;
    for(register int i=head[x];i;i=nxt[i]){
        int u=to[i];if(u==pre) continue;
        for(register int j=siz[x];j;j--)
            for(register int k=1;k<=j;k++)
                dp[x][j]=max(dp[x][j],dp[u][k]+dp[x][j-k]-val[i]);
    }
}

int main(){
    memset(dp,-0x3f,sizeof(dp));
    n=rd(),m=rd();
    for(register int i=1;i<=n-m;i++){
        int k=rd();
        for(register int j=1;j<=k;j++){
            int x=rd(),y=rd();
            add(i,x,y),add(x,i,y);
        }
    }
    for(register int i=n-m+1;i<=n;i++) w[i]=rd();
    dfs(1,0);
//  for(register int i=1;i<=n;i++)
//      for(register int j=1;j<=n;j++)
//          printf("dp[%d][%d]=%d
",i,j,dp[i][j]); 
    for(register int i=siz[1];i;i--)
        if(dp[1][i]>=0) {ans=i;break;}
    cout<<ans<<endl;
    return 0;
}