AtCoder ABC201-BのWAの理由を教えて下さい。

テストケースの2/10がWAになります。正答率からしてNが極小か極大で間違っていると推測はしているのですが手詰まりになってしまいましたため、どこが間違っているか教えていただけますでしょうか？

python3
1
2def merge_sort(aList):
3    if len(aList) <= 1:
4        return aList
5
6    mid = len(aList) // 2
7    left = aList[:mid]
8    right = aList[mid:]
9
10    left = merge_sort(left)
11    right = merge_sort(right)
12
13    return merge(left, right)
14
15
16def merge(left, right):
17    sorted_list = []
18    left_index = 0
19    right_index = 0
20
21    while left_index < len(left) and right_index < len(right):
22        if left[left_index][1] <= right[right_index][1]:
23            sorted_list.append(left[left_index])
24            left_index += 1
25        else:
26            sorted_list.append(right[right_index])
27            right_index += 1
28
29    if left:
30        sorted_list.extend(left[left_index:])
31    if right:
32        sorted_list.extend(right[right_index:])
33
34    return sorted_list
35
36
37N = int(input())
38L = []
39A = [list(input().split()) for _ in range(N)]
40for i in range(N):
41  L.append([A[i][0], int(A[i][1])])
42  
43print(merge_sort(L)[-2][0])

行動規範の内容に同意します

回答2件

自己解決

int型にせずにソートしていたのがいけなかったようです。また、marge_sort()のreturnをlistになっていたため、最終outputのindexがきちんと指定できていませんでした。

python3
1def merge_sort(aList):
2    if len(aList) <= 1:
3        return aList
4
5    mid = len(aList) // 2
6    left = aList[:mid]
7    right = aList[mid:]
8
9    left = merge_sort(left)
10    right = merge_sort(right)
11
12    return merge(left, right)
13
14
15def merge(left, right):
16    sorted_list = []
17    left_index = 0
18    right_index = 0
19
20    while left_index < len(left) and right_index < len(right):
21        if left[left_index][1] <= right[right_index][1]:
22            sorted_list.append(left[left_index])
23            left_index += 1
24        else:
25            sorted_list.append(right[right_index])
26            right_index += 1
27
28    if left:
29        sorted_list.extend(left[left_index:])
30    if right:
31        sorted_list.extend(right[right_index:])
32
33    return sorted_list
34
35
36N = int(input())
37L = []
38A = [list(input().split()) for _ in range(N)]
39for i in range(N):
40  L.append([A[i][0], int(A[i][1])])
41  
42
43if N == 2:
44    if L[0][1] > L[1][1]:
45        print(L[1][0])
46    else:
47        print(L[0][0])
48else:
49    print(merge_sort(L)[-2][0])