글 읽기 - 시간초과 어디를 잘라야 할가요..

rntmfxkfcnf 2년 전 0

시간초과가 나는데 어디가 불필요할까요 ?

글읽어 주셔서 감사합니다.

import sys

board = [list(map(int, sys.stdin.readline().split())) for i in range(9)]

arr = []

pos = []

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

_set = set(i for i in range(1, 10))


def fun2(all, c):

    if c == len(pos):
        for i in board:
            print(" ".join(list(map(str, i))))
        exit()

    board[pos[c][0]][pos[c][1]] = all[c]

    for i in range(4):
        ny = pos[c][0] + dy[i]
        nx = pos[c][1] + dx[i]

        while 0 <= ny < 9 and 0 <= nx < 9:
            if all[c] == board[ny][nx]:
                all[c] = 0
                return
            ny += dy[i]
            nx += dx[i]

    fun2(all, c + 1)


def main():

    for h in range(3):
        for w in range(3):
            nums = []
            for i in range(h * 3, (h + 1) * 3):
                for j in range(w * 3, (w + 1) * 3):
                    if not board[i][j]:
                        pos.append([i, j])
                        continue
                    nums.append(board[i][j])
            arr.append(list(_set - set(nums)))


def fun(v, c, index, all):
    global fun2

    if c == len(arr[index]):
        if index == len(arr) - 1:
            fun2(list(map(int, (all + v).split())), 0)
            return
        fun("", 0, index + 1, all + v)

    for x in arr[index]:
        if v.count(str(x)) == 0:
            t = v + str(x) + " "
            fun(t, c + 1, index, all)


main()
fun("", 0, 0, "")

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

문제

문제

출처

ICPC

시간초과 어디를 잘라야 할가요..

rntmfxkfcnf 2년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말