poj 3140(树形dp)

题目链接：http://poj.org/problem?id=3140

思路：简单树形dp题，dp[u]表示以u为根的子树的人数和。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<climits>
#include<algorithm>
#include<stack>
#include<map>
#include<set>
#include<vector>
#include<queue>
#include<list>
using namespace std;
#define MAXN 111111
#define inf 1<<30
#define INF 1LL<<60
typedef long long ll;
typedef pair<int,int>PP;
template<class T> inline T Get_MIN(const T &a,const T &b){ return a < b ? a : b; }
template<class T> inline T Get_MAX(const T &a,const T &b){ return a > b ? a : b; }
template<class T> inline T ABS(const T &a){ return a < 0 ? -a : a; }

struct Edge{
    int v,next;
}edge[MAXN<<4];

int n,m,NE;
int head[MAXN];

void Insert(int u,int v)
{
    edge[NE].v=v;
    edge[NE].next=head[u];
    head[u]=NE++;
}

ll dp[MAXN],sum,MIN;
void dfs(int u,int father)
{
    for(int i=head[u];i!=-1;i=edge[i].next){
        int v=edge[i].v;
        if(v==father)continue;
        dfs(v,u);
        dp[u]+=dp[v];
    }
    MIN=Get_MIN(MIN,ABS((dp[u]-(sum-dp[u]))));
   // cout<<u<<"**"<<dp[u]<<"****"<<abs(sum-dp[u])<<"**"<<MIN<<endl;
}


int main()
{
    int u,v,t=1;
    while(~scanf("%d%d",&n,&m)&&(n||m)){
        NE=0;
        memset(head,-1,sizeof(head));
        sum=0;
        for(int i=1;i<=n;i++)scanf("%lld",&dp[i]),sum+=dp[i];
        while(m--){
            scanf("%d%d",&u,&v);
            Insert(u,v);
            Insert(v,u);
        }
        MIN=INF;
        dfs(1,-1);
        printf("Case %d: %lld
",t++,MIN);
    }
    return 0;
}