글 읽기 - 17144번 파이썬 인덱스 에러 질문입니다.

wh4044 2년 전 0

모든 테스트 케이스는 통과했는데 파이썬으로 돌리면 시간 초과가 나오고,

pypy3로 돌리면 런타임에러(index)라고 뜨는데 뭐가 문제일까요 ㅠ.ㅠ

dx = [-1, 0, 1, 0]
dy = [0, 1, 0, -1]
r, c, t = map(int, input().split())
a = [list(map(int, input().split())) for _ in range(r)]
x1=x2=0
for i in range(r):
    if a[i][0] == -1:
        if x1==0:
            x1 = i
        else:
            x2 = i
 
def micro(x, y, b):    # 미세먼지가 퍼지는 함수
    cnt = 0
    for i in range(4):
        nx = x + dx[i]
        ny = y + dy[i]
        if 0 <= nx < r and 0 <= ny < c and a[nx][ny] != -1:
            if int(b/5) > 0:
                d[nx][ny] += int(b/5)
                cnt += 1    
    a[x][y] = b - (int(b/5))*cnt    

def clean():
    a2 = [0]

    for j in range(1, c):
        a2.append(a[x2][j])
    for i in range(x2+1, r):
        a2.append(a[i][-1])
    for j in range(c-2, -1, -1):
        a2.append(a[-1][j])
    for i in range(r-2, x2, -1):
        a2.append(a[i][0])

    idx = 1
    k2 = len(a2)
    for j in range(1, c):
        a[x2][j] = a2[idx-1]
        idx += 1
        idx %= k2

    for i in range(x2+1, r):
        a[i][-1] = a2[idx-1]
        idx += 1
        idx %= k2
    for j in range(c-2, -1, -1):
        a[-1][j] = a2[idx-1]
        idx += 1
        idx %= k2        
    for i in range(r-2, x2, -1):        
        a[i][0] = a2[idx-1]
        idx += 1
        idx %= k2

    a1 = [0]
    for j in range(1, c):
        a1.append(a[x1][j])
    for i in range(x1-1, -1, -1):
        a1.append(a[i][-1])
    for j in range(c-2, -1, -1):
        a1.append(a[0][j])
    for i in range(1, x1):
        a1.append(a[i][0])

    idx2 = 1
    k1 = len(a1)
    for j in range(1, c):
        a[x1][j] = a1[idx2-1]
        idx2 += 1
        idx2 %= k1
    for i in range(x1-1, -1, -1):
        a[i][-1] = a1[idx2-1]
        idx2 += 1
        idx2 %= k1
    for j in range(c-2, -1, -1):
        a[0][j] = a1[idx2-1]
        idx2 += 1
        idx2 %= k1
    for i in range(1, x1):
        a[i][0] = a1[idx2-1]
        idx2 += 1
        idx2 %= k1

for _ in range(t):
    munzi = []
    d = [[0]*(r+1) for _ in range(c+1)]
    for i in range(r):
        for j in range(c):
            if a[i][j] > 0:
                munzi.append((i, j, a[i][j]))                

    for x, y, b in munzi:
        micro(x, y, b)

    for i in range(r):
        for j in range(c):
            a[i][j] += d[i][j]
    clean()

total = 2
for i in a:
    total += sum(i)

print(total)

ensia96 2년 전 0

87번째 줄에서, 미세먼지의 확산 상태를 저장하는 d 배열을 선언하실 때, r과 c를 잘못 입력하신 것 같네요.

문제에 주어지는 예제가 다 맞는 이유는, 행과 열의 길이 차이가 1이어서 r+1, c+1이어도, 문제가 없기 때문일 거예요.

for _ in range(t):
    munzi = []
    d = [[0]*(r+1) for _ in range(c+1)] # X
    d = [[0]*c for _ in range(r)]       # O

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

문제

문제

출처

ICPC

17144번 파이썬 인덱스 에러 질문입니다.

wh4044 2년 전 0

ensia96 2년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말