pku 3140 Contestants Division(树的划分)

题目：给定一棵树，每个节点有一个权值。任意去掉一条边会形成两颗子树，每颗子树有一个权值和，求二者相差的最小值。

分析：先求出所有节点权值总和，然后以任意一个节点为根节点，深搜，求每个节点的（本身及其子节点）权值和，直接和sum比较就可以得到去除该节点和父节点之间的边时的答案。

教训：又一次栽倒在vector上，因为是多数据，每次使用vector之前都需要clear。因为忘了这个WA了数次。

#include <string> #include <vector> #include <cmath> #include <queue> #include <algorithm> #include <iostream> #define PI 3.14159265358979323846264338327950288 #define _clr(a,b) memset(a,b,sizeof(a)) template<class T> T _abs(T a) { if(a<0) return -a;return a;} template<class T> void get_min(T& a,T b) { if(a>b) a=b;} template<class T> void get_max(T& a,T b) { if(a<b) a=b;} using namespace std; int N,M; int stu[100005]; bool visited[100005]; vector<int> edges[100005]; __int64 sum; __int64 ans; __int64 DFS(int root) { visited[root]=true; __int64 result=stu[root]; for(int i=0;i<edges[root].size();i++) { if(!visited[edges[root][i]]) { result+=DFS(edges[root][i]); } } get_min(ans,_abs(sum-result-result)); return result; } int main() { int cases=1; while(cases) { scanf("%d%d",&N,&M); if(N==0&&M==0) break; sum=0; ans=(__int64)1<<62; for(int i=1;i<=N;i++) edges[i].clear(); for(int i=1;i<=N;i++) { scanf("%d",&stu[i]); sum+=stu[i]; } int start,end; for(int i=0;i<M;i++) { scanf("%d%d",&start,&end); edges[start].push_back(end); edges[end].push_back(start); } _clr(visited,0); DFS(5); printf("Case %d: %I64d/n",cases++,ans); } return 0; }