POJ 3140 Contestants Division 简单树形DP

注意到题目说每两个点之间有且仅有一条路径，由此可得是一棵树。那么接下来就简单了，直接dfs即可。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<vector>
#include<algorithm>
using namespace std;
#define LL long long
#define INF 0x3f3f3f3f3f3f3f3f
const int maxn=1e5+10;
vector<int> G[maxn];
LL w[maxn],sum[maxn];

void dfs(int u,int fa)
{
    int len=G[u].size();
    for(int i=0;i<len;i++)
    {
        int v=G[u][i];
        if(v==fa) continue;
        dfs(v,u);
        sum[u]+=sum[v];
    }
}

int main()
{
    int n,m,u,v,cas=0;;
    while(~scanf("%d%d",&n,&m))
    {
        LL cnt=0;
        if(n==0&&m==0) break;
        for(int i=1;i<=n;i++) scanf("%lld",&w[i]),cnt+=w[i],G[i].clear();
        for(int i=1;i<=n;i++) sum[i]=w[i];
        for(int i=1;i<n;i++)
        {
            scanf("%d%d",&u,&v);
            G[u].push_back(v);
            G[v].push_back(u);
        }
        dfs(1,0);
        LL ans=INF;
        for(int i=1;i<=n;i++) ans=min(ans,llabs(cnt-2*sum[i]));
        printf("Case %d: %lld\n",++cas,ans);
    }
    return 0;
}