pta题集——功夫传人

题目:

一门武功能否传承久远并被发扬光大,是要看缘分的。一般来说,师傅传授给徒弟的武功总要打个折扣,于是越往后传,弟子们的功夫就越弱…… 直到某一支的某一代突然出现一个天分特别高的弟子(或者是吃到了灵丹、挖到了特别的秘笈),会将功夫的威力一下子放大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

题目分析:

以题目给出的样例为例,这个师门的简易谱系图如下图所示。第0行为祖师爷,其有三个徒弟分别为2、3、5,第2、3、5行又分别为编号2、3、5的徒弟,以此类推。

创建一个字典来存储师门的谱系关系,键为师傅的编号,值为徒弟编号的列表。从祖师爷(编号0)开始遍历师门谱系。在遍历过程中,记录当前的功力值,每向下传一代,根据折扣百分比r计算新的功力值(即当前功力值乘以(1−r/100))。创建一个字典用来记录得道者及其功力放大倍数,键为得道者的编号,值为功力放大倍数。当遇到 “得道者”(Ki​=0)时,将当前功力值乘以其武功放大倍数,并累加到结果中。最后,输出所有 “得道者” 的功力总值的整数部分。

输入处理和数据容器初始化:

n, z, r = map(float, input().split())
n = int(n)
dic = {}
dic_get = {}

n为师门人数,z为祖师爷功力值,r为折扣百分比值

字典dic用来记录师门的谱系关系

字典dic_get用来记录得道者的编号及其功力放大倍数

读取师门谱系信息:

for i in range(n):
    ls = list(map(int, input().split()))
    if ls[0] != 0:
        dic[i] = ls[1:]
    else:
        dic_get[i] = ls[1]

ls存储读取到的每一行的信息,ls[0]即为该行的人物(编号i)的徒弟个数,若ls[0]为0,则该行的人物(编号i)为得道者。若ls[0]不为0则将编号i及其徒弟编号的列表存储进字典dic中,否则将编号i的人物及其功力放大倍数存储进字典dic_get中。

使用栈模拟深度优先搜索:

stack = [(0, z)]
getsum = 0

while stack:
    person, power = stack.pop()
    if person in dic_get:
        getsum += power * dic_get[person]
    else:
        new_power = power * (1 - r / 100)
        if person in dic:
            for apprentice in dic[person]:
                stack.append((apprentice, new_power))

stack是一个栈,初始时将祖师爷的编号 0 和其功力值 z 作为一个元组 (0,z)压入栈中。

getsum 用于累计所有 “得道者” 的功力总值,初始值为 0

 

进入 while 循环,只要栈不为空,就继续循环。person, power = stack.pop() 从栈中弹出一个元素,分别得到当前人物的编号 person 和其功力值 power。若 person 在 dic_get 中,说明该人物是 “得道者”,将其功力值 power 乘以放大倍数 dic_get[person],并累加到 getsum 中。若 person 不在 dic_get 中,说明该人物不是 “得道者”,计算下一代的功力值 new_power = power * (1 - r / 100)。若 person 在 dic 中,说明该人物有徒弟,将其每个徒弟的编号和新的功力值作为元组压入栈中。

 

知识拓展:

这里运用到了数据结构中栈的知识。

栈是一种特殊的线性表,其特点是仅允许在一端进行插入和删除操作。这一端被称为栈顶,而另一端称为栈底。由于这种特性,栈遵循后进先出的原则

如果有对栈这部分知识不了解的同学可以看一下这位博主的博客:数据结构详解:栈的定义、操作及应用-CSDN博客

 

错误分析:

博主之前使用的方法是用dfs深度优先算法,定义了一个dfs函数。但使用这个方法进行测试时测试点出现了非零返回的情况,于是只好改为用栈模拟深度优先算法。

 

整体代码展示:

n, z, r = map(float, input().split())
n = int(n)
dic = {}
dic_get = {}

for i in range(n):
    ls = list(map(int, input().split()))
    if ls[0] != 0:
        dic[i] = ls[1:]
    else:
        dic_get[i] = ls[1]

stack = [(0, z)]
getsum = 0

while stack:
    person, power = stack.pop()
    if person in dic_get:
        getsum += power * dic_get[person]
    else:
        new_power = power * (1 - r / 100)
        if person in dic:
            for apprentice in dic[person]:
                stack.append((apprentice, new_power))

print(int(getsum))

 

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值