2011.计算机院.Problem C.寻找宝藏

题目描述
有一棵多叉树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;
}