PAT (Basic Level) Practice (中文)1018 锤子剪刀布 python (无超时 完全通过)

出处:
https://pintia.cn/problem-sets/994805260223102976/problems/994805304020025344

1018 锤子剪刀布 (20 分) 大家应该都会玩“锤子剪刀布”的游戏:两人同时给出手势,胜负规则如图所示:
PAT (Basic Level) Practice (中文)1018 锤子剪刀布 python (无超时 完全通过)_第1张图片
现给出两人的交锋记录,请统计双方的胜、平、负次数,并且给出双方分别出什么手势的胜算最大。

输入格式:

输入第 1 行给出正整数 N(≤10​^5​​ ),即双方交锋的次数。随后 N 行,每行给出一次交锋的信息,即甲、乙双方同时给出的的手势。C
代表“锤子”、J 代表“剪刀”、B 代表“布”,第 1 个字母代表甲方,第 2 个代表乙方,中间有 1 个空格。

输出格式:

输出第 1、2 行分别给出甲、乙的胜、平、负次数,数字间以 1 个空格分隔。第 3 行给出两个字母,分别代表甲、乙获胜次数最多的手势,中间有
1 个空格。如果解不唯一,则输出按字母序最小的解。 输入样例:

10 C J J B C B B B B C C C C B J B B C J J 输出样例:

5 3 2 2 3 5 B B

开始按照题目的要求写出来的代码是这样的:
超时代码:

n = int(input())
#n为比赛次数

AC,AJ,AB,BC,BJ,BB = 0,0,0,0,0,0
#A,B代表甲和乙,AC,AJ,AB分别为甲锤,剪刀,包袱赢的次数,BC,BJ,BB同理

victoryA,victoryB,loseA,loseB,equalA,equalB = 0,0,0,0,0,0,
#victoryA(B),loseA(B),equalA(B)分为为甲(乙)胜、输、平的次数

for i in range(n):
    res = input().split()
    #res[0]、res[1]分别为当前一局甲乙出的手势
    if res[0] == res[1]:
    #首先判断是否为平局
        equalA += 1
        equalB += 1
    #然后按剪子包袱锤规则判定胜负,因为前面已经判断过,所以以后不用考虑平局
    elif res[0] == 'J':
        if res[1] == 'B':
            victoryA += 1
            loseB += 1
            AJ += 1
        else:
            victoryB += 1
            loseA += 1
            BC += 1
    elif res[0] == 'B':
        if res[1] == 'J':
            victoryB += 1
            loseA += 1
            BJ += 1
        else:
            victoryA += 1
            loseB += 1
            AB += 1
    elif res[0] == 'C':
        if res[1] == 'J':
            victoryA += 1
            loseB += 1
            AC += 1
        else:
            victoryB += 1
            loseA += 1
            BB += 1

#判断那种手势赢的次数最多,因为次数相同按字母顺序前的输出,所以前面的大于等于后面即可,后面必须大于前面
if AB >= AC and AB >= AJ:
        resA = 'B'
elif AC > AB and AC >= AJ:
    resA = 'C'
elif AJ > AB and AJ > AC:
    resA = 'J'

if BB >= BC and BB >= BJ:
        resB = 'B'
elif BC > BB and BC >= BJ:
    resB = 'C'
elif BJ > BB and BJ > BC:
    resB = 'J'

#按格式输出
print(victoryA, equalA, loseA)
print(victoryB, equalB, loseB)
print(resA, resB)

逻辑上应该没问题,但是会有一个测试显示超时。我看很多人写的这个题的代码也是出现了最后一个测试点超时的问题。
PAT (Basic Level) Practice (中文)1018 锤子剪刀布 python (无超时 完全通过)_第2张图片

应该是数据量大的时候运行时间太长了,python经常会出现这个问题。
怎么进行优化呢?我想到之前看到过的:
导入sys,用sys.stdin.readline()代替input()
于是导入sys后把:

res = input().split()

换成:

res = sys.stdin.readline().split()

代码:

import sys

n = int(input())
#n为比赛次数

AC,AJ,AB,BC,BJ,BB = 0,0,0,0,0,0
#A,B代表甲和乙,AC,AJ,AB分别为甲锤,剪刀,包袱赢的次数,BC,BJ,BB同理

victoryA,victoryB,loseA,loseB,equalA,equalB = 0,0,0,0,0,0,
#victoryA(B),loseA(B),equalA(B)分为为甲(乙)胜、输、平的次数

for i in range(n):
    res = sys.stdin.readline().split()
    #res[0]、res[1]分别为当前一局甲乙出的手势
    if res[0] == res[1]:
    #首先判断是否为平局
        equalA += 1
        equalB += 1
    #然后按剪子包袱锤规则判定胜负,因为前面已经判断过,所以以后不用考虑平局
    elif res[0] == 'J':
        if res[1] == 'B':
            victoryA += 1
            loseB += 1
            AJ += 1
        else:
            victoryB += 1
            loseA += 1
            BC += 1
    elif res[0] == 'B':
        if res[1] == 'J':
            victoryB += 1
            loseA += 1
            BJ += 1
        else:
            victoryA += 1
            loseB += 1
            AB += 1
    elif res[0] == 'C':
        if res[1] == 'J':
            victoryA += 1
            loseB += 1
            AC += 1
        else:
            victoryB += 1
            loseA += 1
            BB += 1

#判断那种手势赢的次数最多,因为次数相同按字母顺序前的输出,所以前面的大于等于后面即可,后面必须大于前面
if AB >= AC and AB >= AJ:
        resA = 'B'
elif AC > AB and AC >= AJ:
    resA = 'C'
elif AJ > AB and AJ > AC:
    resA = 'J'

if BB >= BC and BB >= BJ:
        resB = 'B'
elif BC > BB and BC >= BJ:
    resB = 'C'
elif BJ > BB and BJ > BC:
    resB = 'J'

#按格式输出
print(victoryA, equalA, loseA)
print(victoryB, equalB, loseB)
print(resA, resB)

本来没抱什么希望,没想到完美通过。
PAT (Basic Level) Practice (中文)1018 锤子剪刀布 python (无超时 完全通过)_第3张图片

看来在数据量大的时候导入sys读取数据还是很有必要的。

你可能感兴趣的:(PAT,python,python)