POJ3140 - 树的DFS

  题目抽象成模型就是给了N个点...每个点都有权值..知道这N个点连线的关系...问..去掉一条边...使两边点的总权之差最小....不会出现环路...每条边都是双向的...

  其实这个图就是一棵树...开始找这题是想练树型DP...但发现这题直接DFS就可以了...

 .做一次DFS...得到每个子树的权之和.( 每个点的sum对应的是这个点为根的子树的所有点全之和)...然后所有扫一遍...所有点权之和为totol...找到 abs ( totol - sum[ i ] )最小的..就是答案...

  几个细节一定要注意...

  首先...这个树给的边不是单向的....所以存的时候不能简单的用father[ ] 来存....只能先存下一个图...然后令任意点为根开始DFS....

  数据量比较大....算totol和sum[ ] 的时候都有可能超过 int 范围...所以要开 long long ... 读入输出用 %I64d 

  题目所给的m为n的十倍...但稍微思考下树的性质...可以知道...树的边最多为点的个数 -1 ...但由于这个题预先不知道方向....所以开 2n 大小的空间存线就够了

Program :

#include<iostream> 
#include<math.h>
#include<stdio.h>
#define MAX  99999999999999999ll
#define MAXN 100001
using namespace std;
struct pp
{
    int to,next;      
}line[MAXN*3];
int n,m,t,v[MAXN];
__int64 sum[MAXN],totol,ans; 
bool used[MAXN];
__int64 absl(__int64 a)
{
    if (a<0) return -a; return a;
}
void DFS(int num)
{
    int k=v[num]; 
    used[num]=true;   
    while (k)
    {
       if (!used[line[k].to])
       {
           DFS(line[k].to);           
           sum[num]+=sum[line[k].to];               
       }
       k=line[k].next;
    }  
    return;
}
void GetAnswer()
{
    int i; 
    ans=MAX; 
    memset(used,false,sizeof(used));
    DFS(1);
    for (i=1;i<=n;i++) 
      if (ans>absl(totol-2*sum[i]))
        ans=absl(totol-2*sum[i]);
    return;
}
int main()
{ 
    t=0;
    while (scanf("%d%d",&n,&m)!=EOF)
    {
        if (!n && !m) break;  
        memset(v,0,sizeof(v));
        totol=0;
        for (int i=1;i<=n;i++) 
        {
            scanf("%I64d",&sum[i]);
            totol+=sum[i];
        }
        for (int i=1;i<=m;i++) 
        {
           int x,y;
           scanf("%d%d",&x,&y);
           line[i*2-1].next=v[x]; line[i*2-1].to=y; v[x]=i*2-1;
           line[i*2].next=v[y]; line[i*2].to=x; v[y]=i*2;  
        }    
        t++; 
        GetAnswer();
        printf("Case %d: %I64d\n",t,ans);         
    }  
    return 0;   
}