【PAT-L2-020】功夫传人

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

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

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

输入格式：

输入在第一行给出3个正整数，分别是：N（<=10⁵）——整个师门的总人数（于是每个人从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为零表示这是一位得道者，这时后面跟的一个数字表示其武功被放大的倍数。

输出格式：

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

输入样例：

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

比赛时用找父节点的做法，最后一组一直TLE啊啊啊 ╭(╯^╰)╮

如果从每个传道者开始向上找父亲的话，最坏的复杂度N^N

正确的做法从0点开始向下搜，记录层数的信息 bfsdfs都可以，注意0如果是传道者需要特判，就这样。

想通了后感觉自己好水啊o(╥﹏╥)o~什么时候才能像泰神一样优秀呢

dfs

 #include <bits/stdc++.h>
 using namespace std;
 typedef long long LL;
 const int N = ;
 int bei[N];
 double sum = , z, r;

 vector<int>V[N];
 void dfs(int s, int ceng)
 {
     for(unsigned int i = ; i < V[s].size(); i++)
     {
         int v = V[s][i];
         if(bei[v])
         {
             sum += bei[v]*z*pow(-r*0.01, ceng);
             continue;
         }
         dfs(v, ceng+);
     }
 }
 int main()
 {
     int n, m, a;
     scanf("%d%lf%lf", &n, &z, &r);
     for(int i = ; i < n; i++)
     {
         scanf("%d", &m);
         if(m==)
         {
             scanf("%d", &a);
             bei[i] = a;
             continue;
         }
         while(m--)
         {
             scanf("%d", &a);
             V[i].push_back(a);
         }
     }
     if(bei[]) sum += bei[]*z;
     dfs(, );
     LL ans = (int)sum;
     printf("%lld\n", ans);
     return ;
 }

bfs

 #include <bits/stdc++.h>
 using namespace std;
 typedef long long LL;
 const int N = ;
 int bei[N];
 double sum = , z, r;
 struct node
 {
     int id, ceng;
 };
 vector<int>V[N];
 void bfs()
 {
     queue<node>Q;
     node s, t;
     s.id = , s.ceng = ;
     Q.push(s);
     if(bei[]) sum += bei[]*z;
     while(!Q.empty())
     {
         s =  Q.front();
         Q.pop();
         for(unsigned int i = ; i < V[s.id].size(); i++)
         {
             t.id = V[s.id][i];
             t.ceng = s.ceng+;
             if(bei[t.id])
                 sum += bei[t.id]*z*pow(-r*0.01, t.ceng);
             Q.push(t);
         }
     }
 }
 int main()
 {
     int n, m, a;
     scanf("%d%lf%lf", &n, &z, &r);
     for(int i = ; i < n; i++)
     {
         scanf("%d", &m);
         if(m==)
         {
             scanf("%d", &a);
             bei[i] = a;
             continue;
         }
         while(m--)
         {
             scanf("%d", &a);
             V[i].push_back(a);
         }
     }
     bfs();
     LL ans = (int)sum;
     printf("%lld\n", ans);
     return ;
 }