HDOJ树形DP专题之Centroid

题目链接

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

View Code

 1 #include <stdio.h>
 2 #include <string.h>
 3 #include <vector>
 4 #define N 16000
 5 #define MAX(a,b) ((a)>(b)?(a):(b))
 6 using namespace std;
 7 vector<int> g[N];
 8 int n,p[N],d[N],sum[N],w[N],dmax,ans[N],cnt;
 9 void dfs(int u,int fa)
10 {
11     int i,v;
12     d[u]=(fa==-1?0:d[fa]+1);
13     dmax=MAX(dmax,d[u]);
14     for(i=0; i<g[u].size(); i++)
15     {
16         v=g[u][i];
17         if(v!=fa) dfs(v,p[v]=u);
18     }
19 }
20 void dp()
21 {
22     int i,j;
23     memset(w,0,sizeof(w));
24     for(i=0; i<n; i++)  sum[i]=1;
25     for(i=dmax; i>=0; i--)
26     {
27         for(j=0; j<n; j++)
28         {
29             if(d[j]!=i) continue;
30             w[j]=MAX(w[j],n-sum[j]);
31             if(i>0) w[p[j]]=MAX(w[p[j]],sum[j]),sum[p[j]]+=sum[j];
32         }
33     }
34 }
35 int main()
36 {
37     int i,t,u,v,min;
38     scanf("%d",&n);
39     for(i=0; i<n; i++)  g[i].clear();
40     for(i=0; i<n-1; i++)
41     {
42         scanf("%d%d",&u,&v);
43         u--,v--;
44         g[u].push_back(v);
45         g[v].push_back(u);
46     }
47     dmax=0;
48     dfs(0,-1);
49     dp();
50     min=0x7fffffff;
51     cnt=0;
52     for(i=0; i<n; i++)
53     {
54         if(w[i]<min)  cnt=0,min=w[i],ans[cnt++]=i;
55         else if(w[i]==min)  ans[cnt++]=i;
56     }
57     printf("%d %d\n",min,cnt);
58     for(i=0; i<cnt-1; i++)  printf("%d ",ans[i]+1);
59     printf("%d",ans[cnt-1]+1);
60     return 0;
61 }