第三个点开O2re，不开wa，求大佬帮下忙

#include<iostream>
using namespace std;

const int N=1e4;
struct edge {
	int v, next;
} e[N];
int n, r[N], dp[N][2];
int cnt, head[N];
bool vis[N];

void add(int a, int b) {
	e[++cnt].v=b;
	e[cnt].next=head[a];
	head[a]=cnt;
	return ;
}

void dfs(int u) {
	for (int i=head[u]; i; i=e[i].next) {
		int to=e[i].v;
		if (vis[to]) continue;
		vis[to]=true;
		dfs(to);
		dp[u][0]+=max(dp[to][1], dp[to][0]);
		dp[u][1]+=dp[to][0];
	}
	dp[u][1]+=r[u];
	return ; 
}

int main() {
	int x, y;
	cin >> n;
	for (int i=1; i<=n; i++)
		cin >> r[i];
	while (1) {
		cin >> x >> y;
		if (x==0 && y==0) break;
		add(x, y);
		add(y, x);
	}
	vis[1]=true;
	dfs(1);
	cout << max(dp[1][0], dp[1][1]);
	return 0;
}