【机试备考】Day22-寻找宝藏 | 树

题目

BUPT 2011 计算机 ProblemC（oj）
有一棵多叉树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

题解

思路：

1. 记录每一个节点的孩子数量和父节点序号
2. 顺着L的父亲往上爬，根据每一次父亲的孩子数量计算概率

#include <bits/stdc++.h>
using namespace std;
int main()
{
    
    
    int n,m,l,a,b;
    cin>>n>>m>>l;
    double res=1;
    vector<int>child(1001,0);//记录孩子数量
    vector<int>father(1001,-1);//记录父子关系
    while(m--)
    {
    
    
        cin>>a>>b;
        child[a]++;//a的孩子+1
        father[b]=a;//b的父节点是a
    }
    while(father[l]!=-1)
    {
    
    
        l=father[l];//往上爬
        res/=child[l];
    }
    printf("%.6f\n",res);
}