题目描述
有一棵多叉树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
#include<bits/stdc++.h>
using namespace std;
int t[1005];
int dp[1005];
int main() {
double sum=1;
int N,M,L,a,b;
cin>>N>>M>>L;
memset(t,-1,sizeof(t));
memset(dp,0,sizeof(dp));
while(M--){
cin>>a>>b;//a->b
t[b]=a;
dp[a]++;
}
while(t[L]!=-1){
L=t[L];
sum*=dp[L];
}
printf("%lf",1.0/sum);
return 0;
}