글 읽기 - 파이썬 잘못된곳 지적 부탁드립니다.

leehojin0719 3년 전 0

안녕하세요

이 문제 풀다가 계속 오류가 나서 어느부분이 잘못된건지 알고싶어 글 쓰게 되었습니다.

먼저 장애물을 세 개 세우고 다 세웠으면 bfs함수를 실행하여 T가 일직선으로 채워지게 하였습니다.

그리고 학생의 수를 세서 값이 3이 나오면 return하게 하였는데 실행하면 시간도 오래걸리고 값이 제대로 나오질 않네요

지적해주시면 감사하겠습니다.

혹시나 주석이 이해안되시는 분은 댓글 남겨주세요

from collections import deque

n = int(input())

graph = []

for i in range(n):
    graph.append(input().split())
    
temp = [[0] * n for i in range(n)]#복사받을 리스트

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

def bfs(x, y, tx, ty):#tx와 ty를 이용해서 원래의 위치를 기억
    q = deque()
    q.append((x, y))
    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 >= n:#범위를 벗어나면
                continue
            if nx == tx or ny == ty:#새로운 nx, ny가 기존 T와 일직선에 위치하면
                if temp[nx][ny] == 'X' or temp[nx][ny] == 'S': #X또는 S라면
                    temp[nx][ny] = 'T'#T로 바꿈
                    q.append((nx, ny))
            else:
                continue
                
def check():#학생의 수를 세는 함수
    count = 0
    for i in range(n):
        for j in range(n):
            if temp[i][j] == 'S':
                count += 1
    return count



result = 0

def fence(fen):#장애물을 세 개 만들고 bfs를 적용하는 함수
    global result
    if fen == 3:#장애물(fen)을 세 개 만들었을 때
        for i in range(n):
            for j in range(n):
                temp[i][j] = graph[i][j]#temp에 복사
        for i in range(n):
            for j in range(n):
                if temp[i][j] == 'T':#만약 T면 bfs함수 실행
                    bfs(i, j, i, j)
        if check() == 3:#학생 세 명이 있을때
            return print('YES')
        
    for i in range(n): #장애물(fen)이 3개 이하일 때
        for j in range(n):
            if graph[i][j] == 'X':
                graph[i][j] = 'O'
                fen += 1
                fence(fen)
                graph[i][j] = 'X'
                fen -= 1
                
fence(0)#fence 함수에 0을 넣음

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

문제

문제

출처

ICPC

파이썬 잘못된곳 지적 부탁드립니다.

leehojin0719 3년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말