글 읽기 - 20058 질문입니다 (

cksl0830 1년 전 0

디버깅 결과 bfs부분에서 3*3 행렬까지만 돌고 더이상 탐색을 멈추는데,, 왜그런지 모르겠습니다 제 코드가 어디서 틀린지 도와주세요 ㅠㅠ

.. 얼음의 양부분은 잘 출력이 되고 덩어리부분이 안됩니다! 계속 9가 나와요 ..

from collections import deque

N, Q = map(int, input().split())
graph = [list(map(int, input().split())) for _ in range(2 ** N)]
L = list(map(int, input().split()))

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


def fire_storm(l):
    global graph

    temp = [[0] * 2 ** N for _ in range(2 ** N)]
    for x in range(0, 2 ** N, l):
        for y in range(0, 2 ** N, l):
            section = [[0] * l for _ in range(l)]
            for i in range(l):
                for j in range(l):
                    section[i][j] = graph[x + i][y + j]
            section = rotate(section)
            for i in range(l):
                for j in range(l):
                    temp[x + i][y + j] = section[i][j]
    return temp


def remove(graph):
    temp = []
    for x in range(2 ** N):
        for y in range(2 ** N):
            if graph[x][y]:
                cnt = 0
                for i in range(4):
                    nx = x + dx[i]
                    ny = y + dy[i]
                    if 0 <= nx < 2 ** N and 0 <= ny < 2 ** N:
                        if graph[nx][ny]:
                            cnt += 1
                if cnt < 3:
                    temp.append((x, y))
    for x, y in temp:
        graph[x][y] -= 1


    return graph


def rotate(a):
    a = list(zip(*a[::-1]))
    a = [list(s) for s in a]
    return a


def bfs(x, y):
    global visit
    q = deque()
    cnt = 1
    q.append((x, y))
    while q:
        x, y = q.popleft()
        for i in range(4):
            nx = x + dx[i]
            ny = y + dy[i]
            if 0 <= nx < N and 0 <= ny < N and visit[nx][ny] ==0 and graph[nx][ny]:
                cnt += 1
                visit[nx][ny] = 1
                q.append((nx, ny))

    return cnt


for l in L:
    graph = fire_storm(2 ** l)
    graph = remove(graph)

answer = 0
count = 0
visit = [[0] * (2 ** N) for _ in range(2 ** N)]

print(graph)
for i in range(2 ** N):
    for j in range(2 ** N):
        answer += graph[i][j]
        if graph[i][j] and visit[i][j] == 0:
            visit[i][j]=1
            count=max(count,bfs(i, j))

if count == 1:
    count = 0

print(answer)
print(count)

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

문제

문제

출처

ICPC

20058 질문입니다 (

cksl0830 1년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말