北邮机试 | bupt oj | 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

AC代码：

#include<bits/stdc++.h>
using namespace std;
typedef struct node{
    int fa;
    int sons;
    node(int fa=-1,int sons=0):fa(fa),sons(sons){}
}mulTree;
mulTree tree[1010];
int main()
{
    int n,m,l;
    int son,father;
    scanf("%d%d%d",&n,&m,&l);
    while(m--){
        scanf("%d%d",&father,&son);
        tree[father].sons++;
        tree[son].fa=father;
    }
    int k=l;
    k=tree[k].fa;
    int mul=1;
    while(k!=-1){
        int si = tree[k].sons;
        mul *= si;
        k=tree[k].fa;
    }
    printf("%.6f",1./mul);
    return 0;
}