105. 寻找宝藏
时间限制 1000 ms 内存限制 65536 KB Special Judge
题目描述
有一棵多叉树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;
vector<int>v[1001];
int vis[1001];
int pre[1001];
int n,m,l;
void bfs()
{
queue<int>q;
vis[0] = 1;
q.push(0);
while(!q.empty())
{
int tmp = q.front();
q.pop();
int k = v[tmp].size();
if(tmp == l)
{
return;
}
for(int i = 0; i < k; i++)
{
int to = v[tmp][i];
if(!vis[to])
{
vis[to] = 1;
q.push(to);
pre[to] = tmp;
}
}
}
}
int main()
{
while(~scanf("%d%d%d", &n,&m,&l))
{
memset(vis, 0, sizeof(vis));
memset(pre, -1, sizeof(pre));
for(int i = 0; i < n; i++)
v[i].clear();
for(int i = 0; i < m; i++)
{
int a, b;
scanf("%d%d", &a, &b);
v[a].push_back(b);
}
bfs();
int ff = l;
double pk = 1.0;
while(ff!=-1)
{
ff = pre[ff];
int k = v[ff].size();
if(k == 0)
k = 1.0;
pk *= 1.0/k;
}
printf("%.6lf\n",pk);
}
return 0;
}
/*
6 5 5
0 1
1 3
0 2
2 4
2 5
*/