글 읽기 - (파이썬)14502 연구소 고수님들 귀중한 시간 부탁드립니다...

daniel0801 2년 전 0

혼자 코딩 공부하고 있는 초보입니다.... 코드를 돌리면 new_graph에 원래 graph가 복사가 되지 않아서 돌리면 결과값이 0이 나오는데 deepcopy는 너무 느려서 사용하지 않고 리스트를 복사하려면 어떻게 해야하는지 알려주실 수 있나요? 부탁드립니다...

import sys
from collections import deque
from itertools import combinations
N, M = map(int, sys.stdin.readline().rstrip().split())
graph = []
for _ in range(N):
  graph.append(list(map(int, sys.stdin.readline().rstrip().split())))
ans = []
def bfs():
  q = deque()
  for i in range(N):
    for j in range(M):
      if new_graph[i][j] == 2:
        q.append((i, j))
  cnt = 0
  dx = [1, -1, 0, 0]
  dy = [0, 0, 1, -1]
  while q:
    x, y = q.popleft()
    for i in range(4):
      nx = x + dx[i]
      ny = y + dy[i]
      if nx <= -1 or nx >= N or ny <= -1 or ny >= M:
        continue
      if new_graph[nx][ny] == 0:
          new_graph[nx][ny] = 2
          q.append((nx, ny))
  for i in range(N):
    for j in range(M):
      if new_graph[i][j] == 0:
        cnt += 1
  return cnt
data = []
for i in range(N):
  for j in range(M):
    data.append((i, j))
result = list(combinations(data, 3))
for i in range(len(result)):
  new_graph = [[0 for _ in range(M)] for _ in range(N)]
  for i in range(N):
    for j in range(M):
      if graph[i][j] != 0:
        new_graph[i][j] = graph[i][j]
  for j in range(3):
    if new_graph[result[i][j][0]][result[i][j][1]] == 0:
      new_graph[result[i][j][0]][result[i][j][1]] = 1
    else:
      break
  ans.append(bfs())
  for j in range(3):
    new_graph[result[i][j][0]][result[i][j][1]] = 0
print(max(ans))

vmffotltka 2년 전 2

new_graph = [item[:] for item in graph]

이중 리스트 복사할 때 이렇게 하는 게 가장 좋더라구요

daniel0801 2년 전 1

@vmffotltka

확실히 deepcopy보다는 그게 빠른거 같아서 좋은거 같습니다. 시간 내주셔서 감사합니다!

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

문제

문제

출처

ICPC

(파이썬)14502 연구소 고수님들 귀중한 시간 부탁드립니다...

daniel0801 2년 전 0

vmffotltka 2년 전 2

daniel0801 2년 전 1

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말