poj 3140 Contestants Division（树形dp? dfs计数+枚举）

本文出自   http://blog.csdn.net/shuangde800

--------------------------------------------------------------------------------------

题目链接： poj-3140

题目

   给n个节点的带权树，删掉其中一边，就会变成两颗子树，
   求删去某条边使得这这两颗子树的权值之差的绝对值最小。

思路

   直接dfs一次，计算所有子树的权值总和tot[i]
   如果删掉一条边(v, fa)，fa是v的父亲节点，
   那么v子树权值总和为tot[v]，显然另一棵子树的权值总和就是sum-tot[v]，
   最总取最小绝对值即可。
   这题要注意用long long

   其实就是dfs+枚举，想不通为什么有人会把这题列为树形dp?

代码

/**=====================================================
 *   This is a solution for ACM/ICPC problem
 *
 *   @source      : poj-3140 Contestants Division
 *   @description : 树形dp, 计数问题
 *   @author      : shuangde
 *   @blog        : blog.csdn.net/shuangde800
 *   @email       : zengshuangde@gmail.com
 *   Copyright (C) 2013/08/31 14:45 All rights reserved. 
 *======================================================*/
#include 
   
   
    
    
#include 
    
    
     
     
#include 
     
     
      
      
#include 
      
      
       
       
#include 
       
       
         #include 
        
          #include 
         
           using namespace std; typedef long long int64; const int INF = 0x3f3f3f3f; const int MAXN = 100010; namespace Adj { int size, head[MAXN]; struct Node{ int v, next; }E[MAXN*2]; inline void initAdj(){ size = 0; memset(head, -1, sizeof(head)); } inline void addEdge(int u, int v) { E[size].v = v; E[size].next = head[u]; head[u] = size++; } } using namespace Adj; int n, m; int num[MAXN]; int64 ans, sum; int64 Abs(int64 a) { return a<0?-a:a; } int64 dfs(int u, int fa) { int64 tot = num[u]; for (int e = head[u]; e != -1; e = E[e].next) { int v = E[e].v; if (v == fa) continue; tot += dfs(v, u); } ans = min(ans, Abs(sum-tot-tot)); return tot; } int main(){ int cas = 1; while (~scanf("%d%d", &n, &m) && n + m) { sum = 0; for (int i = 1; i <= n; ++i) { scanf("%d", &num[i]); sum += num[i]; } initAdj(); for (int i = 0; i < m; ++i) { int u, v; scanf("%d%d", &u, &v); addEdge(u, v); addEdge(v, u); } ans = sum; dfs(1, -1); cout << "Case " << cas++ << ": " << ans << endl; } return 0; }