POJ 3140 Contestants Division （树形DP，简单）

题意：

　　有n个城市，构成一棵树，每个城市有v个人，要求断开树上的一条边，使得两个连通分量中的人数之差最小。问差的绝对值。（注意本题的M是没有用的，因为所给的必定是一棵树，边数M必定是n-1）

思路：

　　考虑当前节点t，当断开t与父亲的边时，“子树t中的人数”与“剩下的人数”之差的绝对值若最小，则为答案。

//#include <bits/stdc++.h>
#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <map>
#include <deque>
#include <algorithm>
#include <vector>
#include <iostream>
#define pii pair<int,int>
#define max(x,y) ((x)>(y)?(x):(y))
#define min(x,y) ((x)<(y)?(x):(y))
#define abs(x) ((x)<0?-(x):(x))
#define INF 2147483647
#define LL  long long
using namespace std;
const double PI  = acos(-1.0);
const int N=1000010;
int edge_cnt, head[N], w[N];
LL ans, sum;
struct node
{
    int from, to, next;
    node(){};
    node(int from,int to,int next):from(from),to(to),next(next){};
}edge[N*2];

void add_node(int from,int to)
{
    edge[edge_cnt]=node(from,to,head[from]);
    head[from]=edge_cnt++;
}

LL DFS(int t,int far)  //枚举删除t头上的边
{
    node e;
    LL cnt=0;
    for(int i=head[t]; i!=-1; i=e.next)
    {
        e=edge[i];
        if(e.to==far)   continue;
        cnt+=DFS(e.to, t);
    }
    cnt+=w[t];  //本子树的人数
    ans=min(ans, abs( 2*cnt-sum ));
    return cnt;
}


int main()
{
    //freopen("input.txt", "r", stdin);
    int a, b, n, m, Case=0;
    while(scanf("%d%d",&n,&m), n+m)
    {
        sum=edge_cnt=0;
        memset(head,-1,sizeof(head));

        for(int i=1; i<=n; i++)
        {
            scanf("%d",&w[i]);
            sum+=w[i];
        }
        ans=sum;
        for(int i=0; i<m; i++)
        {
            scanf("%d%d",&a,&b);
            add_node(a,b);
            add_node(b,a);
        }
        DFS(1,-1);
        printf("Case %d: %lld
", ++Case, ans);
    }
    return 0;
}