首页 > 代码库 > POJ 2057 - The Lost House
POJ 2057 - The Lost House
设一棵树有两个子树A,B。Pa是房子在A子树上的概率,Pb是房子在B子树上的概率。
先走A子树寻找房子的期望是:
【在A子树上找到房子的路程】*Pa+(【没有在A子树上找到房子的路程(固定的一个值)】+【在B子树上找到房子的路程】)*Pb。
先走B子树寻找房子的期望是:
【在B子树上找到房子的路程】*Pb+(【没有在B子树上找到房子的路程(固定的一个值)】+【在A子树上找到房子的路程】)*Pa。
若这两个值做比较的话,可以化简为:
【没有在A子树上找到房子的路程(固定的一个值)】*Pb【没有在B子树上找到房子的路程(固定的一个值)】*Pa相比较。
Pa=(A子树上的叶子节点数)/(所有的叶子节点数);Pb也是如此。
通过排序就可以设计出在树上的寻找路径了。
——引用来自http://blog.csdn.net/lin375691011/article/details/31446343
当然,由于所有的叶子结点数目是固定的,所以可以直接用A子树上的叶子节点数代替Pa,类似的Pb也是如此。
(真的是一道非常不错的题目,值得深究)
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 }
另外,请注意http://poj.org/showmessage?message_id=341806,这影响到输入时的数据存储方式,之前我是按照:边是按顺序一层层下来的,就一直WA
POJ 2057 - The Lost House
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。