글 읽기 - 연구소

lsk4393 3년 전 0

파이썬으로 작성했는데 어디서 논리적 오류가 생기는지 찾지 못하고 있습니다. 고수분들 도와주세용~

from itertools import combinations
from collections import deque

N, M = map(int, input().split())
arr = [[0 for j in range(M)] for i in range(N)]
copy = [[0 for j in range(M)] for i in range(N)]

dx = [0,0,-1,1]
dy = [1,-1,0,0]

answer =0

for i in range(N):
    row_input = list(map(int,input().split()))
    for j in range(M):
        arr[i][j] = row_input[j]

combi_list = []
virus_list = []

for i in range(N):
    for j in range(M):
        if arr[i][j] == 0 :
            combi_list.append([i,j])
        elif arr[i][j] == 2 :
            virus_list.append([i,j])

blocks = list(combinations(combi_list,3))

def bfs():
    global copy
    global answer
    visit = [[0 for j in range(M)] for i in range(N)]
    dq = deque()
    sum = 0
    for virus in virus_list :
        dq.append([virus[0],virus[1]])
        visit[virus[0]][virus[1]] = 1

    while dq :
        size = len(dq)

        for i in range(0,size):
            virus = dq.popleft()
            x, y = virus[0], virus[1]

            for j in range(0,4):
                next_x = x + dx[j]
                next_y = y + dy[j]

                if next_x < 0 or next_y < 0 or next_x > N-1 or next_y > M-1 :
                    continue
                if copy[next_x][next_y] == 1 or copy[next_x][next_y] == 2:
                    continue
                if visit[next_x][next_y] == 1:
                    continue

                copy[next_x][next_y] = 2
                dq.append([next_x,next_y])
                visit[next_x][next_y] = 1


    for i in range(0,N):
        for j in range(0,M):
            if copy[i][j] == 0 :
                sum+=1

    if answer < sum :
        answer = sum


for block in blocks :
    for wall in block :
        arr[wall[0]][wall[1]] = 1

    copy = arr[:]
    bfs()

    for wall in block :
        arr[wall[0]][wall[1]] = 0

print(answer)

qkrwnghd96 3년 전 0

파이썬 특성상 copy = arr[:] 는 얕은 복사이기 때문에

대신 copy.deepcopy(arr) 을 사용하셔야 합니다.

그리고 모듈의 이름과 변수의 이름이 copy 로 같기 때문에 변수의 이름을 변경해주셔야 해요.

변경한 결과 예제는 돌아가는데, 제출은 하지 않았습니다.

그리고 bfs 에서 queue 를 사용하실 때, size 를 규정하실 필요는 없을 것 같습니다. 41번 째 줄과 43번 째 줄은 사용하지 않으셔도 무방해보입니다.

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

문제

문제

출처

ICPC

연구소

lsk4393 3년 전 0

qkrwnghd96 3년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말