洛谷P1122：最大子树和（树形dp + dfs）

2020.5.21
萌新自从上次区域赛那题碰壁，就励志学好树形dp，先从简单题练练手。

树形结构，前向星自不用说，观察给出的数据，在每一层的子问题是剪枝和不剪枝，那肯定是如果对答案贡献大于零就不剪，小于0就剪掉，所以有如下公式

dp[u] = dp[v] > 0 ? dp[v] : 0

回到问题上，我们的子树可能不止一个，那就加等，最后的答案就是每一层的最优解。

最近好好练练dp，因为还是太重要了。

最近训练有所收获，能够处理一些dp问题了，昨天群里的面试题只花了一分钟就能看出转移公式了，再接再厉吧。

代码：

#include <bits/stdc++.h>
using namespace std;
#define limit (100000 + 5)//防止溢出
#define INF 0x3f3f3f3f
#define inf 0x3f3f3f3f3f
#define lowbit(i) i&(-i)//一步两步
#define EPS 1e-6
#define FASTIO  ios::sync_with_stdio(false);cin.tie(0);
#define ff(a) printf("%lld\n",a );
#define pi(a,b) pair<a,b>
#define rep(i, a, b) for(int i = a; i <= b ; ++i)
#define per(i, a, b) for(int i = b ; i >= a ; --i)
#define mint(a,b,c) min(min(a,b), c)
#define MOD 998244353
#define FOPEN freopen("C:\\Users\\administrator01\\CLionProjects\\untitled24\\data.txt", "rt", stdin)
typedef long long ll;
typedef unsigned long long ull;
ll read(){
    ll sign = 1, x = 0;char s = getchar();
    while(s > '9' || s < '0' ){if(s == '-')sign = -1;s = getchar();}
    while(s >= '0' && s <= '9'){x = x * 10 + s - '0';s = getchar();}
    return x * sign;
}//快读
void write(ll x){
    if(x / 10) write(x / 10);
    putchar(x % 10 + '0');
}
int n , m;
int dp[limit];
struct node{
    int to, next;
}edge[limit<<1];
int head[limit<<1],cnt;
void init(){
    cnt = 0;
    memset(head, -1, sizeof(head));
}
void add(int u, int v){
    edge[cnt].to = v;
    edge[cnt].next = head[u];
    head[u] = cnt++;
}
int f[limit];
int ans;
void dfs(int u, int pre){
    dp[u] = f[u];
    for(int i = head[u]; ~i; i = edge[i].next){
        int v= edge[i].to;
        if(v != pre){
            dfs(v,u);
            dp[u] += dp[v] > 0 ? dp[v] : 0;
        }
    }
    ans = max(ans, dp[u]);
}

int main(){
#ifdef LOCAL
    FOPEN;
    //freopen("C:\\Users\\administrator01\\CLionProjects\\untitled24\\out.txt", "w", stdout);
#endif
    n = read();
    rep(i ,1,n)f[i] = read();
    init();
    ans = 0;
    rep(i ,1, n-1){
        int x = read(), y = read();
        add(x,y);
        add(y,x);
    }
    dfs(1,0);
    write(ans);
    return 0;
}