2839: 石头剪刀布

参考代码:

def generate(x, y):
    flag = True
    while flag:
        for i in range(len(x)):
            if len(y) < N:
                y.append(x[i])
            else:
                flag = False
N, NA, NB = map(int, input().split())
arrA, arr1 = list(map(int, input().split())), []
arrB, arr2 = list(map(int, input().split())), []
sumA, sumB = 0, 0
generate(arrA,arr1)
generate(arrB,arr2)
for i in range(N):
    if arr1[i] == arr2[i]:
        continue
    elif arr1[i] == 2 and arr2[i] == 5 or arr1[i] == 0 and arr2[i] == 2 or arr1[i] == 5 and arr2[i] == 0:
            sumA += 1
    elif arr1[i] == 2 and arr2[i] == 0 or arr1[i] == 0 and arr2[i] == 5 or arr1[i] == 5 and arr2[i] == 2:
            sumB += 1
if sumA > sumB:
    print('A')
elif sumA < sumB:
    print('B')
else:
    print('draw')