poj3140——contestants division

题目大意：一棵树中n个结点，每个结点权值不同，问删掉一条边所形成的两个连通分量各自权值之差最小是多少

输入：结点个数n 边个数m（1 ≤ n ≤ 100000）

           第i个结点的权值（一共n个数，空格隔开）

           s t（一共m行，代表第s个和第t个结点之间有边）

           0 0//代表结束

输出：Case i：最小权值差

分析：树形dfs——算出第i个结点为根的子树的权值之和sum[i]，另一连通分量的即为整棵树的权值和-sum[i]，遍历每个结点即可。注意这棵树不是单向的，即不能存成父子结点的形式，而是在图中双向存边

代码：转载自https://www.2cto.com/kf/201404/291522.html

#include <iostream>
#include <cstring>
#include <cstdlib>
#include <cstdio>
#include <cmath>

using namespace std;

typedef long long LL;

struct edge
{
    int v;//一条边中的后一个结点
    int next;//根节点u的最后一条边的前一条边
} edges[200100];
int count1 = 0;
int head[100100];

void addedge(int u,int v)
{
    edges[count1].v = v;
    edges[count1].next = head[u];
    head[u] = count1++;
}
bool vis[100100];
LL num[100100];
LL ans[100100];
LL out = 0;
int m,n;
LL sum = 0;
LL fabs(LL x)
{
    if(x >= 0)
        return x;
    return -x;
}
bool dfs(int k)
{
    if(vis[k])return false;
    ans[k] += num[k];
    vis[k] = 1;
    for(int i = head[k];i != -1;i = edges[i].next)//从最后一个加入根节点k的边开始遍历
    {
        int t = edges[i].v;
        if(dfs(t))
        ans[k] += ans[t];//ans记录了以k为根的子树的权值和
    }
    out = min(out,fabs(sum-2*ans[k]));//求的是以k为根的树(ans[k])与另外一个连通分量(sum-ans[k])的差的最小值
    return true;
}
int main()
{
  //freopen ("in.txt" , "r" , stdin);
  int t = 0;
  while(scanf("%d%d",&n,&m)==2)
  {
      if(m == 0 && n == 0) break;
      t++;
      sum = 0;
      memset(vis,0,sizeof(vis));
      memset(head,-1,sizeof(head));
      memset(ans,0,sizeof(ans));
      count1 = 0;//记录边的条数
      for(int i = 1;i <= n;i++){
        scanf("%lld",num + i);
        sum += num[i];//整棵树权和
      }
      out = sum;
      for(int i = 0;i < m;i++) {
        int a,b;
        scanf("%d%d",&a,&b);
        addedge(a,b);
        addedge(b,a);
      }
      dfs(1);
      printf("Case %d: %lld\n",t,out);
      return 0;
  }
}