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; }