POJ3140 Contestants Division(树形DP)

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

分析：先求出所有节点权值总和，然后以任意一个节点为根节点，深搜，求每个节点的（本身及其子节点）权值和，直接和总权值比较就可以得到去除该节点和父节点之间的边时的答案，看从哪个节点开始搜索导致两棵子树权值差别最小。

 代码如下：一共WA了19次，最后找到错误的地方，才发现原来是函数的返回值类型不对。

#include <iostream> #include <vector> #include <cmath> #include <queue> #include <algorithm> 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;} using namespace std; __int64 ans,sum; int b[100005],s[100005],n,m; vector<int>v[100005]; void init() { int i,a,b1; for(i=1;i<=n;i++) { scanf("%d",&b[i]); sum+=b[i]; } for(i=1;i<=m;i++) { scanf("%d%d",&a,&b1); /*if(!s[b1]) { v[a].push_back(b1); s[b1]=1; } else { v[b1].push_back(a); s[a]=1; }*/ v[a].push_back(b1); v[b1].push_back(a); } } __int64 dfs(int x) { int i,j; s[x]=1; __int64 re=b[x]; for(i=0;i<v[x].size();i++) if(s[v[x][i]]==0) re+=dfs(v[x][i]); get_min(ans,_abs(sum-re-re)); return re; } int main() { int ncase=1; while(ncase) { scanf("%d%d",&n,&m); if(n==0&&m==0) break; sum=0; int i,j; for(i=1;i<=n;i++) {v[i].clear();s[i]=0;} init();ans=(__int64)1<<62; //memset(s,0,sizeof(s)); dfs(1); printf("Case %d: %I64d/n",ncase++,ans); } return 0; }