글 읽기 - 어떤 문제가 발생하는지는 알겠는데 케이스를 모르겠습니다

rshtiger 3년 전 0

계속해서 런타임에러가 발생하는데, 중간의

# 예외처리 : 배열이 M - 1개로 나누어떨어진 경우 M-1번째 배열에서 가장 큰 수를 M번째 배열에 보낸다
if len(sub_sqnc[-1]) == 0:
sub_sqnc[-1].append(sub_sqnc[-2][-1])
del (sub_sqnc[-2][-1])

이 부분에서 IndexError가 발생하는 것 같습니다.

sub_sqnc가 [ ~~~~, [], [], []] 이런 모양으로 만들어져 오류가 나는 것 같은데

이런 모양이 나오는 경우를 도저히 못 찾겠습니다

그냥 첫 TC부터 오류가 뜨는 것 같은데

어떤 경우에 IndexError가 발생하는걸까요? ㅠㅠ

N, M, K = map(int, input().split())
sqnc = [i for i in range(1, N + 1)]
sub_sqnc = []

sub_size = int(N / M) + (1 if N % M != 0 else 0)

# M이 1인경우 그냥 바로 끝내버린다
if M == 1:
    if N == K:
        sqnc.reverse()
        print(sqnc)
    else:
        print(-1)
    quit(0)

# M개 배열로 나누기
for i in range(0, M - 1):
    sub_sqnc.append(sqnc[sub_size * i:sub_size * (i + 1)])

sub_sqnc.append(sqnc[sub_size * (M - 1):])

# 예외처리 : 배열이 M - 1개로 나누어떨어진 경우 M-1번째 배열에서 가장 큰 수를 M번째 배열에 보낸다
if len(sub_sqnc[-1]) == 0:
    sub_sqnc[-1].append(sub_sqnc[-2][-1])
    del (sub_sqnc[-2][-1])

# 각 배열들을 거꾸로
for sub in sub_sqnc:
    sub.reverse()

lds = []
k = K - 1

while k > 0:
    for i in range(M - 1, -1, -1):
        if k == 0 or len(sub_sqnc[0]) < 2:
            k = 0
            break
        if len(sub_sqnc[i]) < len(sub_sqnc[0]):
            continue
        if len(sub_sqnc[i]) < 2:
            continue
        lds.append(sub_sqnc[i][1])
        del (sub_sqnc[i][1])
        k -= 1

lds.sort()
lds.reverse()
sub_sqnc.append(lds)

mx = -1
for sub in sub_sqnc:
    mx = max(mx, len(sub))

print(sub_sqnc if mx == K or (mx < K and len(sub_sqnc[-1]) == K - 1) else -1)

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

문제

문제

출처

ICPC

어떤 문제가 발생하는지는 알겠는데 케이스를 모르겠습니다

rshtiger 3년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말