PTA习题解析:获奖(Python)

目录

获奖

题干

在某次竞赛中,判题规则是按解题数从多到少排序,在解题数相同的情况下,按总成绩(保证各不相同)从高到低排序,取排名前 60% 的参赛队(四舍五入取整)获奖,请确定某个队能否获奖。

输入格式

首先输入一个正整数 T,表示测试数据的组数,然后是 T 组测试数据。每组测试的第一行输入 1 个整数 n(1 ≤ n ≤ 15)和 1 个字符串 ms(长度小于 10 且不含空格),分别表示参赛队伍总数和想确定是否能获奖的某个队名;接下来的 n 行输入 n 个队的解题信息,每行一个 1 个字符串 s(长度小于 10 且不含空格)和 2 个整数 m,g(0 ≤ m ≤ 10,0 ≤ g ≤ 100),分别表示一个队的队名、解题数、成绩。当然,n 个队名中肯定包含 ms。

输出格式

对于每组测试,若某队能获奖,则输出 “YES”,否则输出 “NO”。引号不必输出。

测试样例

输入

2
3 team001
team001 2 27
team002 2 28
team003 0 7
14 team010
team012 5 52
team013 5 55
team014 5 57
team005 7 77
team002 7 79
team001 7 89
team003 7 78
team014 7 70
team006 7 75
team009 7 84
team007 7 81
team010 5 62
team008 5 53
team011 5 59

输出

YES
NO

解题思路

这道题分别把得分和答题数存在列表或者字典里,然后多次排序或遍历后也能得出结果,但是这种解法并不是很简洁。注意输入的数据是队伍名、队伍答题数和队伍得分,这些属性的相关性是很高的,可以构造一个 Team 类来描述。

class Team(object):
    def __init__(self, name, count, grade):
        self.name = name      #队名
        self.count = count    #答题数
        self.grade = grade    #得分

构造 Team 类以后,对于每一组输入数据可以实例化为一个 Team 对象来存储。为了方便调试,可以重写魔术方法 “def str(self)”,这样在输出 Team 对象时就可以按照我们规定的格式来构造字符串。

def __str__(self):
    return ("name: {}; count: {}; grade: {}".format(self.name, self.count, self.grade))

例如对于测试样例 1,使用 print() 输出 Team 对象的效果如下。
PTA习题解析:获奖(Python)
根据题目要求“取排名前 60% 的参赛队(四舍五入取整)获奖”,也就是说需要按照某个规则(题意)对所有的队伍进行排名。按照这个思路,我们可以重载 Team 类的 “<” 运算符,这样对一个 Team 对象的 list 使用 sorted() 函数排序时,就可以按照我们制定的规则来排序。为了实现这一点,需要重写魔术方法 “def str(self, other)”。

def __lt__(self, other):
    if self.count == other.count:    #答题数相同的情况下比较总分
        return self.grade > other.grade
    else:                            #答题数不同的情况下比较答题数
        return self.count > other.count

为 Team 对象指定了排序的规则后,接下来就很明了了。首先先读取测试点的数据,将每一组数据实例化为一个 Team 对象后,添加入一个 list 中。

testNum = int(input())    #队伍总数
for i in range(testNum):
    teamNum, teamName = input().split(" ")
    teamNum = int(teamNum)

    teams = []    #存放所有 Team 对象的 list
    for j in range(teamNum):
        name, count, grade = input().split(" ")
        aTeam = Team(name, int(count), int(grade))    #实例化 Team 对象
        teams.append(aTeam)

接下来只需要对这个 list 排序,即可将所有队伍按照获奖顺序排列。

'''
print("排序前:")
for team in teams:
    print(team)
'''        
teams = sorted(teams)
'''
print("排序后:")
for team in teams:
    print(team)
'''

此处以测试样例 2 为例,将排名前后的情况输出查看。
PTA习题解析:获奖(Python)
最后只需要遍历列表,如果要求判断的队伍的下标在 list 的前 60%,说明有获奖,输出即可。注意下标是从 0 开始计算的,而且需要使用 round() 函数四舍五入。

for j in range(len(teams)):
    if teams[j].name == teamName:
        if j + 1 <= round(len(teams) * 0.6):
            print('YES')
        else:
            print('NO')
        break

完整代码

class Team(object):
    def __init__(self, name, count, grade):
        self.name = name
        self.count = count
        self.grade = grade
    
    def __str__(self):
        return ("name: {}; count: {}; grade: {}".format(self.name, self.count, self.grade))
    
    def __lt__(self, other):
        if self.count == other.count:
            return self.grade > other.grade
        else:
            return self.count > other.count

testNum = int(input())
for i in range(testNum):
    teamNum, teamName = input().split(" ")
    teamNum = int(teamNum)

    teams = []
    for j in range(teamNum):
        name, count, grade = input().split(" ")
        aTeam = Team(name, int(count), int(grade))
        teams.append(aTeam)

    teams = sorted(teams)
    for j in range(len(teams)):
        if teams[j].name == teamName:
            if j + 1 <= round(len(teams) * 0.6):
                print('YES')
            else:
                print('NO')
            break

参考资料

Python 面向对象编程
Python的富比较方法__lt__、__gt__之间的关联关系分析
python 列表排序方法sort、sorted技巧篇

上一篇:【无标题】


下一篇:PipeCAD - 项目管理