WOJ2153 哨戒炮II

传送门
相邻两个节点会相互影响。可以只考虑它和它的儿子。
它对它父节点的影响会在计算父节点计入。
注意开 $longlong$和读优。

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int maxn=1e5+10;
int Head[maxn],Next[maxn<<1],V[maxn<<1],cnt=0;
int n,m,u,v;
ll g[maxn],d[maxn],f[maxn][2];
inline void add(int u,int v){Next[++cnt]=Head[u],V[cnt]=v,Head[u]=cnt;}
inline int read(){
    int f=1,x=0;char ch=getchar();
    while(!isdigit(ch)){if(ch=='-') f=-1;ch=getchar();}
    while(isdigit(ch)) x=(x<<3)+(x<<1)+ch-'0',ch=getchar();
    return x*f;
}
inline void dfs(int u,int fa){
    f[u][0]=0,f[u][1]=g[u];
    for(int i=Head[u];i;i=Next[i]) if(V[i]!=fa){
        dfs(V[i],u);
    	f[u][0]+=max(f[V[i]][1],f[V[i]][0]);
  	f[u][1]+=max(f[V[i]][0],f[V[i]][1]+d[u]+d[V[i]]);
    }
}
int main(){
    n=read();
    for(int i=1;i<=n;++i) g[i]=read();
    for(int i=1;i<=n;++i) d[i]=read();
    for(int i=1;i<n;++i) u=read(),v=read(),add(u,v),add(v,u);
    dfs(1,0),cout<<max(f[1][0],f[1][1]);
}