POJ 2057 - The Lost House

 1 #include<cstdio>
 2 #include<cstring>
 3 #include<algorithm>
 4 #define MAXN 1005
 5 using namespace std;
 6 int n,num[MAXN],dp[MAXN][2],head[MAXN];
 7 //dp[u][1]：表示u为根的子树上，成功找到房子的步数和 
 8 //dp[u][0]：表示u为根的子树上，找不到房子的步数和 
 9 //head[u]:第一条以u为头节点的边
10 bool worm[1005];
11 struct Edge{
12     int u,v,adj;//起点、终点、同起点的邻边 
13 }edge[MAXN];
14 bool cmp(int a,int b)
15 {
16     return( (dp[a][0]+2)*num[b] < (dp[b][0]+2)*num[a] );
17 }
18 void dfs(int st)
19 {
20     int p=head[st];
21     dp[st][0]=0;
22     dp[st][1]=0;
23     num[st]=0;
24     if(p==-1) num[st]++;//作为根，本身就是叶子结点 
25     else
26     {
27         int cnt=0,group[10];
28         while(p!=-1)
29         {
30             dfs(edge[p].v);
31             dp[st][0]+=dp[(edge[p].v)][0]+2;
32             num[st]+=num[edge[p].v];
33             group[cnt++]=edge[p].v;
34             p=edge[p].adj;
35         }
36         if(worm[st]) dp[st][0]=0;//如果有worm帮忙，知道这里从这里往下不用再找了，没有房子，就不用跑 
37         sort(group,group+cnt,cmp);
38         int tmp=0;
39         for(int i=0;i<cnt;i++)//i表示在第i棵子树下有房子 
40         {
41             dp[st][1]+=dp[group[i]][1]+(tmp+1)*num[group[i]];
42             tmp+=dp[group[i]][0]+2;//走了i子树，没有找到，返回到st 
43         }//这一段for语句非常漂亮，属于解本题的核心思路
44     } 
45     //printf("point %d: dp[][0]=%d dp[][1]=%d\n",st,dp[st][0],dp[st][1]);
46 }
47 int main()
48 {
49     while(scanf("%d",&n) && n!=0)
50     {
51         memset(head,-1,sizeof(head));
52         int from,cnt=1;
53         char tmp[5];
54         scanf("%d%s",&from,tmp);
55         for(int i=2;i<=n;i++)
56         {
57             scanf("%d%s",&from,tmp);
58             worm[i]=(tmp[0]==‘Y‘?1:0);
59             edge[cnt].u=from; 
60             edge[cnt].v=i;
61             edge[cnt].adj=head[from];
62             head[from]=cnt;
63             cnt++;
64         }
65         /*
66         for(int i=1;i<=n-1;i++) printf("edge %d: from=%d to=%d adj=edge%d\n",i,edge[i].u,edge[i].v,edge[i].adj);
67         for(int i=1;i<=n;i++) printf("head %d: edge %d\n",i,head[i]); //检查边是否存储正确 
68         for(int i=2;i<=n;i++) printf("point %d: worm %s\n",i,worm[i]?"exist":"not exist"); //检查worm是否存储正确
69         */
70         dfs(1); 
71         printf("%.4f\n", ((double)dp[1][1])/((double)num[1]) );
72     }
73 }