POJ 3140 Contestants Division 【树形DP】

<题目链接>

题目大意：
给你一棵树，让你找一条边，使得该边的两个端点所对应的两颗子树权值和相差最小，求最小的权值差。

解题分析：

比较基础的树形DP。

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

typedef long long ll;
#define INF 0x7fffffff7fffffff
#define clr(a,b) memset(a,b,sizeof(a))
#define rep(i,s,t) for(int i=s;i<=t;i++)
const int N = 1e5+10;
struct Edge{
    int to,nxt;
}edge[N<<1];
int n,m,cnt,head[N];
ll val[N],dp[N],ans,sum;

ll llabs(ll a){return a>=0?a:(-a);}

void init(){
    cnt=0;sum=0;clr(head,-1);
}
void addedge(int u,int v){
    edge[cnt].to=v,edge[cnt].nxt=head[u];
    head[u]=cnt++;
}
void dfs(int u,int pre){
    dp[u]=val[u];
    for(int i=head[u];~i;i=edge[i].nxt){
        int v = edge[i].to;
        if(v == pre)continue;
        dfs(v,u);
        dp[u]+=dp[v];
    }
    ans=min(ans,llabs(dp[u]-(sum-dp[u])));     //记录最小的差值
}
int main(){
    int ncase=0;
    while(scanf("%d%d",&n,&m)!=EOF,n||m){
        init();
        rep(i,1,n)scanf("%lld",&val[i]),sum+=val[i];
        rep(i,1,m){
            int u,v;scanf("%d%d",&u,&v);
            addedge(u,v);addedge(v,u);
        }
        ans=INF;
        dfs(1,-1);
        printf("Case %d: %lld\n",++ncase,ans);
    }
}

 

 

2019-02-03