PAT天梯赛 L2-020. 功夫传人【DFS】

最新推荐文章于 2024-02-18 19:58:00 发布

原创最新推荐文章于 2024-02-18 19:58:00 发布 · 382 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#DFS

PAT 天梯赛专栏收录该内容

76 篇文章

订阅专栏

本文针对PAT平台上的算法题目L2-020提供了一个详细的解决方案。该方案采用深度优先搜索算法来解决关于树形结构中节点概率计算的问题，并提供了完整的AC代码实现。

题目链接

https://www.patest.cn/contests/gplt/L2-020

思路
从师父开始一层一层往下搜

然后搜到得道者就更新答案

AC代码

#include <cstdio>
#include <cstring>
#include <ctype.h>
#include <cstdlib>
#include <cmath>
#include <climits>
#include <ctime>
#include <iostream>
#include <algorithm>
#include <deque>
#include <vector>
#include <queue>
#include <string>
#include <map>
#include <stack>
#include <set>
#include <numeric>
#include <sstream>
#include <iomanip>
#include <limits>

#define CLR(a) memset(a, 0, sizeof(a))
#define pb push_back

using namespace std;
typedef long long ll;
typedef long double ld;
typedef unsigned long long ull;
typedef pair <int, int> pii;
typedef pair <ll, ll> pll;
typedef pair<string, int> psi;
typedef pair<string, string> pss;

const double PI = 3.14159265358979323846264338327;
const double E = exp(1);
const double eps = 1e-6;

const int INF = 0x3f3f3f3f;
const int maxn = 1e5 + 5;
const int MOD = 1e9 + 7;

ll n;

double z, r;

struct Node
{
    int tot;
    vector <int> v;
}temp;

ld ans = 0.0;

vector <Node> d, da;

void dfs(int x, int cur)
{
    if (d[x].tot == 0)
    {
        ans += z * pow(r, cur) * d[x].v[0];
        return;
    }
    vector <int>::iterator it;
    for (it = d[x].v.begin(); it != d[x].v.end(); it++)
    {
        dfs((*it), cur + 1);
    }
}

int main()
{
    int num;
    scanf("%lld%lf%lf", &n, &z, &r);
    r = (100 - r) * 1.0 / 100;
    for (int i = 0; i < n; i++)
    {
        scanf("%d", &temp.tot);
        temp.v.clear();
        if (temp.tot == 0)
        {
            scanf("%d", &num);
            temp.v.pb(num);
        }
        for (int j = 0; j < temp.tot; j++)
        {
            scanf("%d", &num);
            temp.v.pb(num);
        }
        d.pb(temp);
    }
    dfs(0, 0);
    printf("%.0Lf\n", ans - 0.5);
}