sgu 143 Long Live the Queen 树状DP_sgu

本文链接：https://blog.youkuaiyun.com/woshinannan741/article/details/52214834

题目大意：

给出一棵树找出这个子树价值的最大值

解题思路

简单的树状DP，判断当前节点的子树，如果这个子树大于0就加上，如果小于0就放弃！

AC代码

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int MAXN = 16000 + 500;
const int INF  = 0x3f3f3f3f;
int head[MAXN],e,w[MAXN],dp[MAXN][3];
int ans[MAXN];
struct Edge{
    int v,nxt,w;
}E[MAXN<<1];
void init()
{
    e = 0;
    memset(head,-1,sizeof head);
}
void add(int u,int v)
{
    E[e].v = v;
    E[e].nxt = head[u];
    head[u] = e++;
}
int DFS(int u,int pre)
{
    ans[u] = w[u];
    for(int i=head[u];~i;i=E[i].nxt){
        int tmp = 0;
        int v =E[i].v;
        if(v==pre)continue;
        tmp = DFS(v,u);
        if(tmp>0)  ans[u] += tmp;
    }
    return ans[u];
}
int main()
{
    int N;
    int u,v;
    scanf("%d",&N);
    init();
    for(int i=1;i<=N;i++)
        scanf("%d",&w[i]);
    for(int i=1;i<N;i++){
        scanf("%d%d",&u,&v);
        add(u,v);add(v,u);
    }
    DFS(1,-1);
    int MAX = -INF;
    for(int i=1;i<=N;i++)
        if(ans[i]>MAX) MAX = ans[i];
    printf("%d",MAX);
    return 0;
}