L2-020 功夫传人 (25 分)

L2-020 功夫传人 (25 分)

题目

一门武功能否传承久远并被发扬光大，是要看缘分的。一般来说，师傅传授给徒弟的武功总要打个折扣，于是越往后传，弟子们的功夫就越弱…… 直到某一支的某一代突然出现一个天分特别高的弟子（或者是吃到了灵丹、挖到了特别的秘笈），会将功夫的威力一下子放大N倍 —— 我们称这种弟子为“得道者”。

这里我们来考察某一位祖师爷门下的徒子徒孙家谱：假设家谱中的每个人只有1位师傅（除了祖师爷没有师傅）；每位师傅可以带很多徒弟；并且假设辈分严格有序，即祖师爷这门武功的每个第i代传人只能在第i-1代传人中拜1个师傅。我们假设已知祖师爷的功力值为Z，每向下传承一代，就会减弱r%，除非某一代弟子得道。现给出师门谱系关系，要求你算出所有得道者的功力总值。

输入格式：

输入在第一行给出3个正整数，分别是：N（≤105）——整个师门的总人数（于是每个人从0到N−1编号，祖师爷的编号为0）；Z——祖师爷的功力值（不一定是整数，但起码是正数）；r ——每传一代功夫所打的折扣百分比值（不超过100的正数）。接下来有N行，第i行（i=0,⋯,N−1）描述编号为i的人所传的徒弟，格式为：

K**i ID[1] ID[2] ⋯ ID[K**i]

其中K**i是徒弟的个数，后面跟的是各位徒弟的编号，数字间以空格间隔。K**i为零表示这是一位得道者，这时后面跟的一个数字表示其武功被放大的倍数。

输出格式：

在一行中输出所有得道者的功力总值，只保留其整数部分。题目保证输入和正确的输出都不超过1010。

输入样例：

10 18.0 1.00
3 2 3 5
1 9
1 4
1 7
0 7
2 6 1
1 8
0 9
0 4
0 3

输出样例：

404

思路

方法1

DFS，先存储每个人的老师（通过学生反存储），并用队列存储有哪些人是得道者。之后对每个得道者dfs，这个过程和并查集的祖先更新是一样的思路。

但是不知道为什么，开始不是用队列存储有哪些人是得道者，而是用一个数组存储这些人的武功扩大倍数，遍历数组看哪些不是0就是得道者，一直WA2。换成数组就过了，很奇怪。

方法2

存储每个人的学生，之后从祖先开始dfs，将所有人的武功值更新。如果这个人是得道者，武功值加入答案。没有剪枝，较上一个答案复杂度应该高一些，但是容易实现。

代码

方法1

#include<bits/stdc++.h>
#define INF 0x3f3f3f3f
#define PI acos(-1)
#define pb push_back
using namespace std;
typedef pair<int, int> P;
typedef long long ll;
const int N = 5e5 + 19;
const ll mod = 1e9 + 7;
const double eps = 1e-12;

int par[N];
double num[N];
double vis[N];
double z, r;
int n;
queue<int> que;

double dfs(int x)
{
    if(par[x] == -1)
        return z;
    if(num[x])
        return num[x];
    return num[x] = dfs(par[x]) * r;
}

int main()
{
    cin >> n >> z >> r;
    r = 1.0 - r / 100.0;
    fill(par, par + n, -1);
    for(int i = 0, m, x; i < n; i++)
    {
        cin >> m;
        for(int j = 0; j < m; j++)
        {
            cin >> x;
            par[x] = i;
        }
        if(m == 0)
        {
            cin >> x;
            vis[i] = x;
            que.push(i);
        }
    }
    double sum = 0;
    while(!que.empty())
    {
        sum += dfs(que.front()) * vis[que.front()];
        que.pop();
    }
    cout << (int)sum << endl;
    return 0;
}

方法2

#include<bits/stdc++.h>
#define INF 0x3f3f3f3f
#define PI acos(-1)
#define pb push_back
using namespace std;
typedef pair<int, int> P;
typedef long long ll;
const int N = 5e5 + 19;
const ll mod = 1e9 + 7;
const double eps = 1e-12;

vector<int> vec[N];
double cnt[N];
double sum;
int n;
double z, r;

void dfs(int x, double z)
{
    if(vec[x].size() == 0)
    {
        sum += z * cnt[x];
    }
    for(int i = 0; i < vec[x].size(); i++)
    {
        dfs(vec[x][i], z * r);
    }
}

int main()
{
    cin >> n >> z >> r;
    r = 1.0 - r / 100.0;
    for(int i = 0, m, x; i < n; i++)
    {
        cin >> m;
        for(int j = 0; j < m; j++)
        {
            cin >> x;
            vec[i].pb(x);
        }
        if(m == 0)
        {
            cin >> x;
            cnt[i] = x;
        }
    }
    dfs(0, z);
    cout << (int)sum << endl;
    return 0;
}