P1352 没有上司的舞会

题目描述

某大学有 n 个职员，编号为 1…n。

他们之间有从属关系，也就是说他们的关系就像一棵以校长为根的树，父结点就是子结点的直接上司。

现在有个周年庆宴会，宴会每邀请来一个职员都会增加一定的快乐指数 ri​，但是呢，如果某个职员的直接上司来参加舞会了，那么这个职员就无论如何也不肯来参加舞会了。

所以，请你编程计算，邀请哪些职员可以使快乐指数最大，求最大的快乐指数。

输入格式

输入的第一行是一个整数 n。

第 2 到第 (n+1) 行，每行一个整数，第(i+1) 行的整数表示 i 号职员的快乐指数 ri​。

第 (n+2) 到第 2n 行，每行输入一对整数 l,k，代表 k 是 l 的直接上司。

输出格式

输出一行一个整数代表最大的快乐指数。

输入输出样例

输入 #1复制

7
1
1
1
1
1
1
1
1 3
2 3
6 4
7 4
4 5
3 5

输出 #1复制

5

说明/提示

数据规模与约定

对于 100% 的数据，保证 1≤n≤6×10^3，−128≤ri​≤127，1≤l,k≤n，且给出的关系一定是一棵树。

采用链式前向星存图，由于未指定根节点，so先找到没有父节点的根节点，then，

开二维dp数组，dp[x][0]代表未选此点，dp[x][1]代表选此点，从root开始进行dfs，

先对dp[x][1]这种情况赋初值，then，在用viss数组避免访问父节点的情况下进行dfs，在回溯时

进行dp操作；

若选了dp[x][1]，则子节点y只能选dp[y][0];

若选了dp[x][0]，则子节点y可选max(dp[y][0],dp[y][1]);

上代码

#include<bits/stdc++.h>
using namespace std;
//#pragma GCC optimize(2)
#define  endl '\n'
#define lowbit(x) ((x)&-(x))

typedef long long ll;
ll mod=998244353;
ll ans=0,n1,m1;
ll t,s1,s2,s3,s4,max1=0,min1=100000000,sum=0,n,m,i,j,k,l,r;
ll u,v;
inline int read() {
	bool sym=0;
	int res=0;
	char ch=getchar();
	while(!isdigit(ch))sym |=(ch =='-'),ch=getchar();
	while(isdigit(ch)) res =(res<<3)+(res<<1)+(ch^48),ch=getchar();
	return sym ? -res : res;
}
void print(int x) {
	if(!x)return;
	print(x/10);
	putchar(x%10+'0');
}
int isPrime(int n) {
	float n_sqrt;
	if(n==1) return 0;
	if(n==2 || n==3) return 1;
	if(n%6!=1 && n%6!=5) return 0;
	n_sqrt=floor(sqrt((float)n));
	for(int i=5; i<=n_sqrt; i+=6) {
		if(n%(i)==0 | n%(i+2)==0) return 0;
	}
	return 1;
}
ll dp[108006][2],a[100086],vis[100086],cnt=0,first[100086],viss[100086];
struct node {
	ll to,next;
} edges[200086];
void add(ll u,ll v) {
	edges[++cnt].to =v;
	edges[cnt].next =first[u];
	first[u]=cnt;

}
void dfs(ll x){
	dp[x][1]=a[x];
	dp[x][0]=0;
	viss[x]=1;
	for(ll i=first[x];i;i=edges[i].next ){
		ll y=edges[i].to ;
		if(viss[y]==0){
				dfs(y);
		dp[x][1]+=dp[y][0];
		dp[x][0]+=max(dp[y][0],dp[y][1]);
		}
	
		
	}
	
	
	
	
}
string as,ss;
ll root;
int main() {


	ios::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);
	cin>>n;
	for(i=1; i<=n; i++)
		cin>>a[i];
	for(i=1; i<=n-1; i++) {
		cin>>u>>v;
		add(u,v);
		add(v,u);
		vis[u]++;
	}
	for(i=1;i<=n;i++)
	{
		if(vis[i]==0)
		{
			root=i;
			break;
		}
	}
	
	dfs(root);
	cout<<max(dp[root][1],dp[root][0]);
	return 0;
}