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; }
原文地址:https://www.cnblogs.com/quietus/p/10300026.html
时间: 2024-11-09 03:40:19