pku 3140 Contestants Division(树的划分)_poj - 3140contestants division-优快云博客

本文链接：https://blog.youkuaiyun.com/logic_nut/article/details/4595793

本文介绍了一种利用深度优先搜索(DFS)实现树形动态规划(DP)的方法，旨在找到给定带权树中，任意删除一条边后形成的两棵子树权值之差的最小值。文章提供了完整的C++代码示例，并强调了在处理多组数据时对数据结构进行清理的重要性。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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

分析：先求出所有节点权值总和，然后以任意一个节点为根节点，深搜，求每个节点的（本身及其子节点）权值和，直接和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; }