MZOJ 1132 && LuoGu P2014 选课

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;
}