글 읽기 - 어떤 부분이 문제인지 모르겠습니다

gci787 1년 전 0

24060번 문제는 k번째 저장되는 숫자를 출력해야 되는 문제입니다.

그래서 저는 저장할 때마다 count라는 변수로 저장될 때마다 수를 증가 시켜 k와 같아지면 res 변수에 이를 저장하고

정렬이 끝나면 res를 출력하도록 하였습니다.

res에 값은 초기 값은 -1로 만일 k번째에 숫자가 저장되지 않을 경우 -1을 출력 시켜야 되는 조건을 충족시켰습니다.

해당 코드의 고쳐야 할 점을 알려주시면 감사하겠습니다.

_, k = map(int, input().split())

count = 0
res = -1


def merge_sort(numbers):
    global count, res
    n = len(numbers)
    if n <= 1:
        return

    mid = n // 2
    left_group = numbers[:mid]
    right_group = numbers[mid:]

    merge_sort(left_group)
    merge_sort(right_group)

    left = 0
    right = 0
    now = 0

    while left < len(left_group) and right < len(right_group):
        if left_group[left] < right_group[right]:
            numbers[now] = left_group[left]
            left += 1
            now += 1
            count += 1
            if count == k:
                res = right_group[left - 1]

        else:
            numbers[now] = right_group[right]
            right += 1
            now += 1
            count += 1
            if count == k:
                res = right_group[right - 1]

    while left < len(left_group):
        numbers[now] = left_group[left]
        left += 1
        now += 1
        count += 1
        if count == k:
            res = right_group[left - 1]

    while right < len(right_group):
        numbers[now] = right_group[right]
        right += 1
        now += 1
        count += 1
        if count == k:
            res = right_group[right - 1]


data = list(map(int, input().split()))

merge_sort(data)
print(res)

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

문제

문제

출처

ICPC

어떤 부분이 문제인지 모르겠습니다

gci787 1년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말