记录编号 |
461792 |
评测结果 |
AAAAAAAAAA |
题目名称 |
火车站饭店 |
最终得分 |
100 |
用户昵称 |
胖周zzf |
是否通过 |
通过 |
代码语言 |
C++ |
运行时间 |
0.434 s |
提交时间 |
2017-10-20 18:45:30 |
内存使用 |
2.98 MiB |
显示代码纯文本
#include<cstdio>
#include<vector>
#include<algorithm>
#include<cstring>
using namespace std;
struct{
int c,f;
vector<int>next;
}x[100010];
int d[100010][2]={0},K;
void TREEDP(int rt,int frt){
d[rt][1]=x[rt].c;d[rt][0]=0;
for(int i=0;i<x[rt].next.size();i++){
if(x[rt].next[i]!=frt&&d[x[rt].next[i]][0]==-1){
TREEDP(x[rt].next[i],rt);
}
}
for(int i=0;i<x[rt].next.size();i++){
if(x[rt].next[i]!=frt){
d[rt][1]+=d[x[rt].next[i]][0];
d[rt][0]+=max(d[x[rt].next[i]][0],d[x[rt].next[i]][1]);
}
}
}
int main(){
freopen("profitz.in","r",stdin);
freopen("profitz.out","w",stdout);
memset(d,-1,sizeof(d));
int n;scanf("%d",&n);
for(register int i=1;i<=n;i++)
scanf("%d",&x[i].c);
for(register int i=1,a,b;i<n;i++){
scanf("%d%d",&a,&b);
x[a].next.push_back(b);
x[b].next.push_back(a);
}
TREEDP(1,0);
printf("%d",max(d[1][0],d[1][1]));
return 0;
}