30ptsWA求助QwQ
查看原帖
30ptsWA求助QwQ
399116
LYqwq楼主2022/7/18 10:33
#include <iostream>
using namespace std;
template<typename T=int>
inline T read(){
    T X=0; bool flag=1; char ch=getchar();
    while(ch<'0' || ch>'9'){if(ch=='-') flag=0; ch=getchar();}
    while(ch>='0' && ch<='9') X=(X<<1)+(X<<3)+ch-'0',ch=getchar();
    if(flag) return X;
    return ~(X-1);
}

const int N=6e3+5;
struct edge{
    int to,nxt;
}e[N];
int n,u,v,rt;
int r[N],s[N];
int head[N],top;
int f[N][2];

void add(int u,int v){
    top++;
    e[top].to=v;
    e[top].nxt=head[u];
    head[u]=top;
}

void dfs(int rt){
    f[rt][0]=0,f[rt][1]=r[rt];
    for(int i=head[rt]; i; i=e[i].nxt){
        v=e[i].to;
        dfs(v);
        f[rt][0]+=max(f[v][0],f[v][1]);
        f[rt][1]+=f[v][0];
    }
}

int main(){
    n=read();
    for(int i=1; i<=n; i++)
        r[i]=read();
    for(int i=1; i<n; i++){
        u=read(),v=read();
        add(v,u);
        s[u]=1;
    }
    for(int i=1; i<=n; i++)
        if(!s[i]){
            rt=i;
            break;
        }
    dfs(rt);
    printf("%d\n",max(f[rt][0],f[rt][1]));
    return 0;
}

记录QwQ 谢谢了

2022/7/18 10:33
加载中...