105. 寻找宝藏(bfs)

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
*/