글 읽기 - [파이썬] 꼭 답변부탁드립니다. ㅠㅠ

coding_master 1년 전 0

python으로 하면 시간초과가 뜨고 pypy로 하면 메모리 초과가 뜹니다.ㅠㅠ,어디 부분이 잘못됬는지 가르쳐주세요.

import sys
sys.setrecursionlimit(10**6)
from collections import deque
a,b = map(int,sys.stdin.readline().split())
graph = []
for _ in range(a):
  k = list(map(int,sys.stdin.readline().split()))
  graph.append(k)
tt = True
line = 0
time = 0
for _ in range(10000000):
  result = 0
  time +=1
  for i in range(1,a-1):
    for ii in range(1,b-1):
      if graph[i][ii] > 0:
        num = 0
        dx = [1,-1,0,0]
        dy = [0,0,-1,1]
        for p in range(4):
          nx = i + dx[p]
          ny = ii + dy[p]
          if line <=graph[nx][ny] <= 0:
            num +=1
        value = graph[i][ii] - num
        if value > 0:
          graph[i][ii] = value
        else:
          graph[i][ii] = line-1
  line -= 1
  visited = [[False]*b for _ in range(a)]
  def bfs(x,y):
    q = deque()
    q.append([x,y])
    visited[x][y] = True
    dx = [1,-1,0,0]
    dy = [0,0,-1,1]
    while q:
      r,rr = q.popleft()
      for rrr in range(4):
        nx = r + dx[rrr]
        ny = rr + dy[rrr]
        if graph[nx][ny] > 0 and visited[nx][ny] == False:
          q.append([nx,ny])
          visited[nx][ny] = True
  for n in range(1,a-1):
    for nn in range(1,b-1):
      if graph[n][nn] > 0 and visited[n][nn] == False:
        bfs(n,nn)
        result +=1
  if result >=2:
    print(time)
    break
  elif result == 0:
    print(time)
    break

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

문제

문제

출처

ICPC

[파이썬] 꼭 답변부탁드립니다. ㅠㅠ

coding_master 1년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말