글 읽기 - Index error가 왜 뜰까요...?

jsyoon0926 1년 전 0

모든 테스트 케이스 문제없이 잘 돌아가는데,

아무리 봐도 인덱스 에러가 뜰 만한 곳이 없어 보이는데...

왜 뜨는 지 도저히 모르겠습니다 ㅠ

고수님들 부탁드립니다...!

import sys
import copy

def isIn(i,j,size) :
    H = size[0]
    W = size[1]
    if i < 0 or i >= H :
        return False
    elif j < 0 or j >= W :
        return False
    else :
        return True


def printRoom(room) :
    H = len(room)
    W = len(room[0])
    for i in range(0,H) :        
        for j in range(0,W) :
        
            if room[i][j] == -2 :
                print(' □ ',end='')
            elif room[i][j] == -3 :
                print(' □ ',end='') 
            elif room[i][j] == -1 :
                print('   ',end='') 
            else :
                if room[i][j]//10 == 0 :
                    print(' ',end='')
                if room[i][j]//100 == 0 :
                    print(' ',end='')
                print(room[i][j],end='')

        print()
    print()  

def direction(d) :
    d_i = 0
    d_j = 0
    if d == 1 :
        d_j = 1
    elif d == 2 :
        d_j = -1 
    elif d == 3 :
        d_i = -1
    elif d == 4 :
        d_i = 1
    return d_i,d_j

def calOneWarmer(room,warmer) :
    size = [len(room),len(room[0])]
    warmedOne = copy.deepcopy(room)
    y,x,d = warmer
    d_i,d_j = direction(d)

    n_i,n_j = 0,0

    i_start,j_start = 0,0
    i_end = size[0]
    j_end = size[1]

    i_jump,j_jump = 0,0
    if d_i == 0 :
        j_start = x + 2*d_j
        j_end = j_start + 2*d_j
        j_jump = 2*d_j
        n_i = 2
        i_jump = 2
        
    elif d_j == 0 :
        i_start = y + 2*d_i
        i_end = i_start + 2*d_i
        i_jump = 2*d_i
        n_j = 2
        j_jump = 2
        
    #print(direction(d))
    warmedOne[y+2*d_i][x+2*d_j] = 5

    for n in range(0,4) :
        temperature = 4 - n
        #print(i_start,i_end,j_start,j_end,i_jump,j_jump)
        for i in range(i_start,i_end,i_jump) :

            for j in range(j_start,j_end,j_jump) :

                if warmedOne[i][j] != 0 :
                    f_i = i + 2*d_i
                    l_i = i + 2*d_i - n_i
                    r_i = i + 2*d_i + n_i

                    f_j = j + 2*d_j
                    l_j = j + 2*d_j - n_j
                    r_j = j + 2*d_j + n_j

                    if isIn(f_i,f_j,size) :                        
                        if room[i + d_i][j + d_j] == -1 :
                            warmedOne[f_i][f_j] = temperature
                    if isIn(l_i,l_j,size) :
                        #print(i,j,l_i,l_j,i - d_j,j - abs(d_i),l_i - d_i,l_j - d_j)
                        if room[i - abs(d_j)][j - abs(d_i)] == -1 and room[l_i - d_i][l_j - d_j] == -1 :
                            warmedOne[l_i][l_j] = temperature
                    if isIn(r_i,r_j,size) :
                        if room[i + abs(d_j)][j + abs(d_i)] == -1 and room[r_i - d_i][r_j - d_j] == -1 :
                            warmedOne[r_i][r_j] = temperature
        i_start += 2*d_i
        i_end += 2*d_i
        j_start += 2*d_j
        j_end += 2*d_j

    return warmedOne

def calWholeWarmer(room,warmer) :
    warmedWhole = copy.deepcopy(room)
    for w in warmer :
        temp = calOneWarmer(room,w)

        for i in range(0,len(room)) :
            for j in range(0,len(room[0])) :
                if warmedWhole[i][j] >= 0 :
                    warmedWhole[i][j] += temp[i][j]

    return warmedWhole


def heatRoom(room,heat) :
    for i in range(0,size[0],2) :
        for j in range(0,size[1],2) :
            room[i][j] += heat[i][j]
    return room

def flowAir(room) :
    size = [len(room),len(room[0])]
    flowed = copy.deepcopy(room)
    for i in range(0,size[0],2) :
        for j in range(0,size[1],2) :
            left = j - 2
            up = i - 2
            if isIn(up,j,size) :
                if room[i - 1][j] == -1 :
                    diff = int((room[i][j] - room[up][j])/4)
                    flowed[i][j] -= diff
                    flowed[up][j] += diff

            if isIn(i,left,size) :
                if room[i][j - 1] == -1 :
                    diff = int((room[i][j] - room[i][left])/4)
                    flowed[i][j] -= diff
                    flowed[i][left] += diff
    #printRoom(flowed)
    for i in range(0,size[0],2) :
        for j in range(0,size[1],2) :
            if i == 0 or i == size[0] - 1 or j == 0 or j == size[1] - 1 :
                if flowed[i][j] > 0 :
                    flowed[i][j] -= 1
    return flowed


def checkK (room,locK,K) :
    check = 1
    for realK in locK :
        i,j = realK
        if room[i][j] < K :
            check = 0
    return check

sys.stdin = open('data.txt','r')
R,C,K = map(int,sys.stdin.readline().split())

room = []
locK = []
warmer = []

for i in range(0,R) :
    room.append([])
    temp = list(map(int,sys.stdin.readline().split()))
    for j in range(0,C) :
        room[2*i].append(0)
        if j != C-1 :
            room[2*i].append(-1)

        if temp[j] == 5 :
            locK.append([2*i,2*j])
        elif temp[j] != 0 :
            warmer.append([2*i,2*j,temp[j]])
        
    if i != R-1 :
        room.append([])
        for j in range(0,C) :
            room[2*i+1].append(-1)
            if j != C-1 :
                room[2*i+1].append(-3)
size = [len(room),len(room[0])]

W = int(sys.stdin.readline())
if W != 0 :
    for w in range(0,W) :    
        y,x,d = map(int,sys.stdin.readline().split())
        i = y-1
        j = x-1
        if d == 0 :
            room[2*i-1][2*j] = -2
        elif d == 1:
            room[2*i][2*j+1] = -2

heat = calWholeWarmer(room,warmer)
ans = 0
while not(checkK(room,locK,K)) :
    room = heatRoom(room,heat)
    #printRoom(room)
    room = flowAir(room)
    ans+= 1
    #printRoom(room)
    if(ans > 100) :
        break
print(ans)

jsyoon0926 1년 전 0

87번 라인에

if isIn(i,j,size) :

를 추가하지 않아서 생긴 오류였습니다. 해결했어요!

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

문제

문제

출처

ICPC

Index error가 왜 뜰까요...?

jsyoon0926 1년 전 0

jsyoon0926 1년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말