글 읽기 - 1201문제 메모리 초과 질문입니다.

skaqn01 5년 전 0

백준을 하면서 메모리 초과는 거의 처음 떠보는거 같아서 당황스럽네요

먼저 이중배열로 문제를 풀었었는데요, 이중배열부분이 문제가 되는거 같아 아래의 방식으로 문제를 풀었습니다. 먼저 groupCountList에 각각의 배열 사이즈를 넣어주면서 풀었는데 중간 메모리를 줄이다. 이정도 변수에도 메모리가 초과되는 걸 보면서 차라리 더 줄이는것 보다 소스를 공개해서 근본적인 원인을 알고 더 좋은 방향으로 코딩습관을 고치고 싶습니다. 도움을 주시면 감사하겠습니다. ㅠㅠ

N, M, K = map(int, input().split())
groupCountList = []
if M + K -1 <= N <= M*K:
    groupCountList.append(K)
    N -= K
    eachCount = N // (M-1)
    remindCount = N % (M-1)
    while(N):
        N -= N // (M-1)
        if remindCount > 0 :
            remindCount -= 1
            groupCountList.append(eachCount + 1)
        else:
            groupCountList.append(eachCount)
    end = 0
    result = ""
    for i in range(len(groupCountList)):
        end += groupCountList[i]
        for i in range(groupCountList[i]):
            result += str(end-i) + " "
    print(result)
else:
    print(-1)

skaqn01 5년 전 0

해결하였습니다.

append를 없애고 미리 배열을 선언하고 풀었습니다. 그래도 이참에 제 코딩습관을 고치고 싶은데 밑의 코드를 보면서 고칠 부분 알려주시면 감사하겠습니다 ㅎㅎ

N, M, K = map(int, input().split())
groupCountList = [0 for _ in range(M)]
if M + K -1 <= N <= M*K:
    groupCountList[0] = K
    N -= K
    if M > 1:
        eachCount = N
        remindCount = N % (M-1)
    else:
        eachCount = N
        remindCount = 0
    idx = 1
    while(N > 0 and  idx < M):
        N -= N
        if remindCount > 0 :
            remindCount -= 1
            groupCountList[idx] = eachCount + 1
            idx += 1
        else:
            groupCountList[idx] = eachCount
            idx += 1
    end = 0
    result = ""
    for i in range(len(groupCountList)):
        end += groupCountList[i]
        for i in range(groupCountList[i]):
            result += str(end-i) + " "
    print(result)
else:
    print(-1)

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

문제

문제

출처

ICPC

1201문제 메모리 초과 질문입니다.

skaqn01 5년 전 0

skaqn01 5년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말