HDOJ树形DP专题之Contestants Division

题目链接

题目大意：给定一棵树，每个结点有一个权值，一棵树的权值为所有结点的权值和，现将这棵树分为两棵子树，要使得两子树的权值差最小。

我的做法是先将无根树化为有根树，然后求每棵子树的权值，最后用一次扫描求结果。需要注意的是结果要用long long型。

#include <stdio.h>
#include <string.h>
#include <vector>
#define N 100000
#define MAX(a,b) ((a)>(b)?(a):(b))
#define MIN(a,b) ((a)<(b)?(a):(b))
#define D(a,b) ((a)>(b)?(a)-(b):(b)-(a))
using namespace std;
vector<int> g[N],dep[N];
int n,m,a[N],d[N],p[N],dmax;
long long sum[N],ans;
void make_set()
{
  for(int i=0;i<n;i++)  p[i]=i;
}
int find_set(int i)
{
  return i==p[i]?p[i]:(p[i]=find_set(p[i]));
}
void dfs(int u,int fa)
{
  int i,v,l;
  d[u]=(fa==-1?0:d[fa]+1);
  dmax=MAX(dmax,d[u]);
  dep[d[u]].push_back(u);
  l=g[u].size();
  for(i=0;i<l;i++)
  {
    v=g[u][i];
    if(v!=fa) dfs(v,p[v]=u);
  }
}
void dp()
{
  int i,j,l,v;
  memset(sum,0,sizeof(sum));
  for(i=dmax;i>=0;i--)
  {
    l=dep[i].size();
    for(j=0;j<l;j++)
    {
      v=dep[i][j];
      sum[v]+=a[v];
      if(i>0) sum[p[v]]+=sum[v];
    }
  }
}
int main()
{
  int i,u,v,kase=0;
  while(scanf("%d%d",&n,&m)&&n)
  {
    for(i=0;i<n;i++)  scanf("%d",&a[i]);
    for(i=0;i<n;i++)  g[i].clear();
    make_set();
    for(i=0;i<m;i++)
    {
      scanf("%d%d",&u,&v);
      u--,v--;
      if(find_set(u)==find_set(v))  continue;
      p[v]=u;
      g[u].push_back(v);
      g[v].push_back(u);
    }
    for(i=0;i<n;i++)  dep[i].clear();
    dmax=0;
    dfs(0,-1);
    dp();
    ans=sum[0];
    for(i=1;i<n;i++)  ans=MIN(ans,D(sum[i],sum[0]-sum[i]));
    printf("Case %d: %lld\n",++kase,ans);
  }
  return 0;
}