全WA求条
查看原帖
全WA求条
1070309
BUS_109_159_18楼主2024/12/28 12:13
#include<bits/stdc++.h>
#define IO ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
const int N=16002;
using namespace std;
vector<int>v[N];
int dp[N][2],n,l,k,r[N];
void dfs(int x,int fa){
	dp[x][0]=0;
	dp[x][1]=r[x];
	for(int y:v[x]){
		if(y==fa)continue;
		dfs(y,x);
		dp[x][1]+=dp[y][0];
		dp[x][0]+=max(dp[y][0],dp[y][1]);
	}
}
int main(){
	memset(dp,-2147483647,sizeof(dp));
	IO;
	cin>>n;
	for(int i=1;i<=n;++i)cin>>r[i];
	for(int i=1;i<n;++i){
		cin>>l>>k;
		v[l].push_back(k);
		v[k].push_back(l);
	}
	int root=1;
	dfs(root,0);
	cout<<max(dp[root][1],dp[root][0]);
	return 0;
}
2024/12/28 12:13
加载中...