HDOJ树形DP专题之Centroid

题目链接

这题跟Balance Act那题差不多，求图的质点。我直接将那题改了一下提交，结果PE了一次，又WA了一次，最后发现是单case，多case的提交为什么WA呢？

#include <stdio.h>
#include <string.h>
#include <vector>
#define N 16000
#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,ans[N],cnt;
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",&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;
    cnt=0;
    for(i=0; i<n; i++)
    {
        if(w[i]<min)  cnt=0,min=w[i],ans[cnt++]=i;
        else if(w[i]==min)  ans[cnt++]=i;
    }
    printf("%d %d\n",min,cnt);
    for(i=0; i<cnt-1; i++)  printf("%d ",ans[i]+1);
    printf("%d",ans[cnt-1]+1);
    return 0;
}