题目
BUPT 2011 计算机 ProblemC(oj)
有一棵多叉树T,你在树根处,宝藏在某一叶子节点L。现在你沿着树枝向叶子方向走去,从不回头,如果遇到树叉节点,你等概率地挑选一个分支继续走。请问,在给定T,L的情况下,你有多大概率拿到宝藏?
输入描述
第一行,整数N,M,L(1< N< 1000,0< M< 1000,0< L< N),分别代表树T上的节点数、树枝的个数,宝藏所在节点。树根为0号结点。
然后有M行,每行两个整数A,B(0≤ A,B< N)代表从节点A到节点B有一条树枝。可以假设节点A总是更靠近树根。
输出描述
所求的概率,输出四舍五入到6位小数,然后换行。
示例
输入
6 5 5
0 1
1 3
0 2
2 4
2 5
输出
0.250000
题解
思路:
- 记录每一个节点的孩子数量和父节点序号
- 顺着L的父亲往上爬,根据每一次父亲的孩子数量计算概率
#include <bits/stdc++.h>
using namespace std;
int main()
{
int n,m,l,a,b;
cin>>n>>m>>l;
double res=1;
vector<int>child(1001,0);//记录孩子数量
vector<int>father(1001,-1);//记录父子关系
while(m--)
{
cin>>a>>b;
child[a]++;//a的孩子+1
father[b]=a;//b的父节点是a
}
while(father[l]!=-1)
{
l=father[l];//往上爬
res/=child[l];
}
printf("%.6f\n",res);
}