HDOJ树形DP专题之Balancing Act

题目链接

题目大意：给定一棵树，树的每个结点有一个权值，每个结点的权值=去掉该结点后剩余的分支中结点最多的那个分支的结点数。求树中权值最小的结点。

这题CE了两次，第一次因为使用了memset没包含头文件，第二次是因为GNU C++没有头文件<memory.h>，使用<string.h>就AC了。

#include <stdio.h>
#include <string.h>
#include <vector>
#define N 20000
#define MAX(a,b) ((a)>(b)?(a):(b))
using namespace std;
vector<int> g[N];
int n,p[N],d[N],sum[N],w[N],dmax;
void dfs(int u,int fa)
{
  int i,v;
  d[u]=(fa==-1?0:d[fa]+1);
  dmax=MAX(dmax,d[u]);
  for(i=0;i<g[u].size();i++)
  {
    v=g[u][i];
    if(v!=fa) dfs(v,p[v]=u);
  }
}
void dp()
{
  int i,j;
  memset(w,0,sizeof(w));
  for(i=0;i<n;i++)  sum[i]=1;
  for(i=dmax;i>=0;i--)
  {
    for(j=0;j<n;j++)
    {
      if(d[j]!=i) continue;
      w[j]=MAX(w[j],n-sum[j]);
      if(i>0) w[p[j]]=MAX(w[p[j]],sum[j]),sum[p[j]]+=sum[j];
    }
  }
}
int main()
{
  int i,t,u,v,min;
  scanf("%d",&t);
  while(t--)
  {
    scanf("%d",&n);
    for(i=0;i<n;i++)  g[i].clear();
    for(i=0;i<n-1;i++)
    {
      scanf("%d%d",&u,&v);
      u--,v--;
      g[u].push_back(v);
      g[v].push_back(u);
    }
    dmax=0;
    dfs(0,-1);
    dp();
    min=0x7fffffff;
    for(i=0;i<n;i++)  if(w[i]<min)  min=w[u=i];
    printf("%d %d\n",u+1,min);
  }
  return 0;
}