题目描述
某大学有N个职员,编号为1~N。他们之间有从属关系,也就是说他们的关系就像一棵以校长为根的树,父结点就是子结点的直接上司。现在有个周年庆宴会,宴会每邀请来一个职员都会增加一定的快乐指数Ri,但是呢,如果某个职员的上司来参加舞会了,那么这个职员就无论如何也不肯来参加舞会了。所以,请你编程计算,邀请哪些职员可以使快乐指数最大,求最大的快乐指数。
输入输出格式
输入格式:
第一行一个整数N。(1<=N<=6000)
接下来N行,第i+1行表示i号职员的快乐指数Ri。(-128<=Ri<=127)
接下来N-1行,每行输入一对整数L,K。表示K是L的直接上司。
最后一行输入0 0
输出格式:
输出最大的快乐指数。
输入输出样例
输入样例#1:
7 1 1 1 1 1 1 1 1 3 2 3 6 4 7 4 4 5 3 5 0 0
输出样例#1:
5
-------------------------------------------
最基础的树形DP
f[i][0]表示不选以i为根的子树不选i,f[i][1]选i
其实不一定按子树划分,只要分成一些连通块就行了
【关于树】:用了邻接链表,用vector存孩子也可以
#include<iostream> #include<cstdio> #include<cstring> using namespace std; const int N=6005; int n,l,k,r[N]; struct edge{ int ne,v; }e[N]; int h[N],cnt=0; void ins(int f,int s){ cnt++; e[cnt].v=s; e[cnt].ne=h[f]; h[f]=cnt; } int f[N][2]; int dp(int i,int flag){ int &ans=f[i][flag]; if(ans!=-1) return ans; ans=0; for(int j=h[i];j;j=e[j].ne){ int v=e[j].v; if(flag==0) ans+=max(dp(v,1),dp(v,0)); if(flag==1) ans+=dp(v,0); } if(flag==1) ans+=r[i]; return ans; } bool son[N];int root=-1; int main(){ scanf("%d",&n); for(int i=1;i<=n;i++) scanf("%d",&r[i]); for(int i=1;i<=n-1;i++){ scanf("%d%d",&l,&k); ins(k,l);son[l]=1; } memset(f,-1,sizeof(f)); for(int i=1;i<=n;i++) if(son[i]==0){root=i;break;} cout<<max(dp(root,0),dp(root,1)); }