글 읽기 - 계수 정렬을 활용한 풀이 (색인 오류 질문)

betaf1sh 3년 전 0

계수 정렬을 활용해 정렬 이후 색인 위치를 정제하여 출력을 했습니다.

34 번 라인의 index 에서 색인에 실패하는 것 같습니다.

index 가 입력되는 케이스의 수여서 색인이 나지 않을 것 같은데 이상하네요 🤔

이 부분이나 다른 블록에서 제가 놓친 반례가 있을까요?

from typing import List


def counting_sort(L: List[int], MAX: int):
    temp = [0]*(MAX+1)
    answer = [None]*(MAX+1)
    for num in L:
        temp[num] += 1
    for index in range(2, MAX+1):
        temp[index] += temp[index-1]
    for index in range(len(L)-1, -1, -1):
        item = L[index]

        answer[temp[item]] = index
        temp[item] -= 1
    return answer[1:]


def test_counting_sort():
    assert counting_sort([1, 3, 4, 3, 5], 200)[:5] == [0, 1, 3, 2, 4]


if __name__ == "__main__":
    ages = []
    names = []
    for __ in range(int(input())):
        age, name = input().split()
        ages.append(int(age))
        names.append(name)
    sorted_index = counting_sort(ages, 200)
    for index in sorted_index:
        if index == None:
            continue
        print(ages[index], names[index])

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

계수 정렬을 활용한 풀이 (색인 오류 질문)

betaf1sh 3년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말