MZOJ 1132 && LuoGu P2014 選課 [傳送門]

#include<bits/stdc++.h>
#define maxn 100000
#define maxm 500
using namespace std;

int k=0,head[maxn];
int f[maxm][maxm],w[maxn];
int ans=0;
int N,M;

struct node{
    int v,nxt;
}e[maxn<<1];

void adde(int u,int v){
    e[k].v=v;
    e[k].nxt
 
=head[u];
    head[u]=k++;
} 

int dp(int u,int fa){
    f[u][0]=0;f[u][1]=w[u];
    int num=1;
    for(int i=head[u];~i;i=e[i].nxt){
        int v=e[i].v;
        
        if(v==fa){
            continue;
        }
        
        int cnt=dp(v,u);//算出子樹的節點個數 
        num+=cnt;//加到總數當中，num包括了根節點 
 

        
        for(int j=num;j>0;j--){//0-1背包的線性優化，所以必須逆序倒推 
            for(int c=0;c<j;c++){
                if(c>cnt)continue;
                else f[u][j]=max(f[u][j],f[u][j-c]+f[v][c]);
            }
        } 
    }
    return num;
}

void init(){
    freopen("in.txt","r",stdin);
    
 
//freopen("out.txt","w",stdout);
}

void readdata(){
    int u;w[0]=0;
    memset(head,-1,sizeof(head));
    scanf("%d%d",&N,&M);
    M++;//要連一個虛擬的根節點0，所以要++ 
    for(int i=1;i<=N;i++){
        scanf("%d%d",&u,&w[i]);
        adde(u,i);
        adde(i,u);
    }
}

void work(){
    dp(0,-1);
    printf("%d",f[0][M]);
}

int main(){
    //init();
    readdata();
    work();
    return 0;
}

MZOJ 1132 && LuoGu P2014 選課