洛谷 P1352 没有上司的舞会

洛谷 P1352 没有上司的舞会

题目描述

某大学有N个职员，编号为1~N。他们之间有从属关系，也就是说他们的关系就像一棵以校长为根的树，父结点就是子结点的直接上司。现在有个周年庆宴会，宴会每邀请来一个职员都会增加一定的快乐指数Ri，但是呢，如果某个职员的上司来参加舞会了，那么这个职员就无论如何也不肯来参加舞会了。所以，请你编程计算，邀请哪些职员可以使快乐指数最大，求最大的快乐指数。
输入格式：
第一行一个整数N。(1<=N<=6000)
接下来N行，第i+1行表示i号职员的快乐指数Ri。(-128<=Ri<=127)
接下来N-1行，每行输入一对整数L,K。表示K是L的直接上司。
最后一行输入0 0
输出格式：
输出最大的快乐指数。
输入输出样例
输入样例
7
1
1
1
1
1
1
1
1 3
2 3
6 4
7 4
4 5
3 5
0 0
输出样例
5

题目分析：这里有位大佬分析的很全面，我就不再分析了。QWQ
https://www.luogu.org/blog/mak2333/sxdp001

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<cstring>
#include<queue>
using namespace std;
struct arr{
	int nd,nx;
}bot[6005];
int head[6005],fa[6005],vis[6005],que[6005],hap[6005];
int dp[6005][2];
int n,s,t,cnt;
queue<int>q;
inline int read(){
	int x=0,w=1;char ch;
	while(ch!='-'&&(ch<'0'||ch>'9')) ch=getchar();
	if(ch=='-') w=-1,ch=getchar();
	while(ch>='0'&&ch<='9') x=(x<<3)+(x<<1)+(ch-48),ch=getchar();
	return x*w;
}
inline void add(int a,int b){bot[++cnt].nd=b;bot[cnt].nx=head[a];head[a]=cnt;}
void bfs(int s){
	q.push(s);
	vis[s]=1;que[++t]=s;
	while(!q.empty()){
		int u=q.front();
		q.pop();
		for(register int i=head[u];i;i=bot[i].nx)
			if(!vis[bot[i].nd]){
				vis[bot[i].nd]=0;
				q.push(bot[i].nd);
				que[++t]=bot[i].nd;
			}
	}
	return ;
}
int main(){
	freopen("P1352.in","r",stdin);
	freopen("P1352.out","w",stdout);
	n=read();
	for(register int i=1;i<=n;++i){fa[i]=i;hap[i]=read();}
	for(register int i=1;i<=n;++i){
		int x=read(),y=read();
		if(x==0&&y==0) break;
		fa[x]=y;add(y,x);
	}
	s=1;
	while(s!=fa[s]) s=fa[s];
	bfs(s);
	for(register int i=t;i>=0;--i) {
		int u=que[i];
		for(register int j=head[u];j;j=bot[j].nx) {
			int v=bot[j].nd;
			dp[u][0]+=max(dp[v][0],dp[v][1]);
			dp[u][1]=max(dp[v][0]+dp[u][1],dp[u][1]);
		}
		dp[u][1]+=hap[u];
	}
	printf("%d\n",max(dp[s][0],dp[s][1]));
	return 0;
}