代碼虛擬空間
函數內部空間存儲在棧中 C++默認棧空間大小為4M 故開在函數內部容易暴棧
全局變量靜態變量存儲在堆中

樹形DP AcWing 285. 沒有上司的舞會

原題鏈接

AcWing 285. 沒有上司的舞會

算法標簽

動態規劃 樹形DP

思路

狀態計算

代碼

#include<bits/stdc++.h>
#define int long long
#define rep(i, a, b) for(int i=a;i<b;++i)
#define Rep(i, a, b) for(int i=a;i>b;--i)
using namespace std;
const int N = 6005;
int a[N], e[N], h[N], ne[N], idx;
int f[N][2];
bool p[N];
inline int read(){
   int s=0,w=1;
   char ch=getchar();
   while(ch<'0'||ch>'9'){if(ch=='-')w=-1;ch=getchar();}
   while(ch>='0'&&ch<='9') s=s*10+ch-'0',ch=getchar();
   return s*w;
}
void put(int x) {
    if(x<0) putchar('-'),x=-x;
    if(x>=10) put(x/10);
    putchar(x%10^48);
}
void add(int a, int b){
    e[idx]=b;
    ne[idx]=h[a];
    h[a]=idx++;
}
void dfs(int u){
    f[u][1]=a[u];
    for(int i=h[u]; i!=-1; i=ne[i]){
        int j=e[i];
        dfs(j);
        f[u][1]+=f[j][0];
        f[u][0]+=max(f[j][0], f[j][1]);
    }
}
signed main(){
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    int n=read();
    rep(i, 1, n+1){
        a[i]=read();
    }
    memset(h, -1, sizeof h);
    rep(i, 0, n-1){
        int a=read(), b=read();
        add(b, a);
        p[a]=true;
    }
    int root=1;
    // 查詢根節點
    while(p[root]){
        root++;
    }
    dfs(root);
    printf("%lld", max(f[root][0], f[root][1]));
    return 0;
}

原創不易
轉載請標明出處
如果對你有所幫助 別忘啦點贊支持哈