【PAT-L2-020】功夫传人
链接:https://www.patest.cn/contests/gplt/L2-020
一门武功能否传承久远并被发扬光大,是要看缘分的。一般来说,师傅传授给徒弟的武功总要打个折扣,于是越往后传,弟子们的功夫就越弱…… 直到某一支的某一代突然出现一个天分特别高的弟子(或者是吃到了灵丹、挖到了特别的秘笈),会将功夫的威力一下子放大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的人所传的徒弟,格式为:
Ki ID[1] ID[2] ... ID[Ki]
其中Ki是徒弟的个数,后面跟的是各位徒弟的编号,数字间以空格间隔。Ki为零表示这是一位得道者,这时后面跟的一个数字表示其武功被放大的倍数。
输出格式:
在一行中输出所有得道者的功力总值,只保留其整数部分。题目保证输入和正确的输出都不超过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
比赛时用找父节点的做法,最后一组一直TLE啊啊啊 ╭(╯^╰)╮
如果从每个传道者开始向上找父亲的话,最坏的复杂度N^N
正确的做法从0点开始向下搜,记录层数的信息 bfsdfs都可以,注意0如果是传道者需要特判,就这样。
想通了后感觉自己好水啊o(╥﹏╥)o~什么时候才能像泰神一样优秀呢
dfs
1 #include <bits/stdc++.h>
2 using namespace std;
3 typedef long long LL;
4 const int N = 100005;
5 int bei[N];
6 double sum = 0, z, r;
7
8 vector<int>V[N];
9 void dfs(int s, int ceng)
10 {
11 for(unsigned int i = 0; i < V[s].size(); i++)
12 {
13 int v = V[s][i];
14 if(bei[v])
15 {
16 sum += bei[v]*z*pow(1-r*0.01, ceng);
17 continue;
18 }
19 dfs(v, ceng+1);
20 }
21 }
22 int main()
23 {
24 int n, m, a;
25 scanf("%d%lf%lf", &n, &z, &r);
26 for(int i = 0; i < n; i++)
27 {
28 scanf("%d", &m);
29 if(m==0)
30 {
31 scanf("%d", &a);
32 bei[i] = a;
33 continue;
34 }
35 while(m--)
36 {
37 scanf("%d", &a);
38 V[i].push_back(a);
39 }
40 }
41 if(bei[0]) sum += bei[0]*z;
42 dfs(0, 1);
43 LL ans = (int)sum;
44 printf("%lld\n", ans);
45 return 0;
46 }
bfs
1 #include <bits/stdc++.h>
2 using namespace std;
3 typedef long long LL;
4 const int N = 100005;
5 int bei[N];
6 double sum = 0, z, r;
7 struct node
8 {
9 int id, ceng;
10 };
11 vector<int>V[N];
12 void bfs()
13 {
14 queue<node>Q;
15 node s, t;
16 s.id = 0, s.ceng = 0;
17 Q.push(s);
18 if(bei[0]) sum += bei[0]*z;
19 while(!Q.empty())
20 {
21 s = Q.front();
22 Q.pop();
23 for(unsigned int i = 0; i < V[s.id].size(); i++)
24 {
25 t.id = V[s.id][i];
26 t.ceng = s.ceng+1;
27 if(bei[t.id])
28 sum += bei[t.id]*z*pow(1-r*0.01, t.ceng);
29 Q.push(t);
30 }
31 }
32 }
33 int main()
34 {
35 int n, m, a;
36 scanf("%d%lf%lf", &n, &z, &r);
37 for(int i = 0; i < n; i++)
38 {
39 scanf("%d", &m);
40 if(m==0)
41 {
42 scanf("%d", &a);
43 bei[i] = a;
44 continue;
45 }
46 while(m--)
47 {
48 scanf("%d", &a);
49 V[i].push_back(a);
50 }
51 }
52 bfs();
53 LL ans = (int)sum;
54 printf("%lld\n", ans);
55 return 0;
56 }