[poj3140]Contestants Division树形dp

题意：切掉树上的某条边，使分开的两棵树上各点的权值和差值最小。

与hdu2196不同的是，此题是点权，其他无太大差别，注意数据范围。

先求出每个节点的子树权值和，然后自底向上dp即可。取$min (abs(sum - 2*dp[u]))$

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cstdlib>
#include<cmath>
#include<iostream>
using namespace std;
typedef long long ll;
const int maxn=1e6+3;
const ll inf=1e17+3;
ll head[maxn],tot,n,m,sum;
struct edge{
    ll to;
    ll nxt;
    ll w;
}e[maxn<<2];
void add_edge(int u,int v,int w){
    e[tot].to=v;
    e[tot].w=w;
    e[tot].nxt=head[u];
    head[u]=tot++;
}

ll dp[maxn<<2],cnt[maxn<<2];

void dfs(ll u,ll fa){
    for(int i=head[u];i!=-1;i=e[i].nxt){
        int v=e[i].to;
        if(v==fa) continue;
        dfs(v,u);
        cnt[u]+=1ll*cnt[v];
    }
    dp[u]=min(dp[u],sum-2*cnt[u]>=0?sum-2*cnt[u]:2*cnt[u]-sum);
}

inline int read(){
    char k=0;char ls;ls=getchar();for(;ls<'0'||ls>'9';k=ls,ls=getchar());
    int x=0;for(;ls>='0'&&ls<='9';ls=getchar())x=(x<<3)+(x<<1)+ls-'0';
    if(k=='-')x=0-x;return x;
}

int main(){
    int k=0;
    while(scanf("%lld%lld",&n,&m)!=EOF&&(n||m)){
        fill(dp+1,dp+n+1,inf);
        memset(head,-1,sizeof head);
        tot=0;
        sum=0;
        int a,b;
        for(int i=1;i<=n;i++){
            cnt[i]=read();
            sum+=1ll*cnt[i];
        }
        for(int i=0;i<m;i++){
            a=read();
            b=read();
            add_edge(a,b,1);
            add_edge(b,a,1);
        }
        printf("Case %d: ",++k);
        dfs(1,-1);
        ll ans=inf;
        for(int i=1;i<=n;i++){
            ans=min(ans,dp[i]);
        }
        printf("%lld
",ans);
    }
    return 0;
}