글 읽기 - N queens 시간초과 질문드립니다!!

alscks0614 1년 전 1

답은 맞게 나오는데 시간 초과가 나옵니다...

행 방향은 1차원 배열로 해서 최적화 된 것 같고,

열 방향도 [x for x in queen if x not in A] 사용해서 지워준 것 같은데

대각선 방향이 문제일까요?

import sys
N=int(sys.stdin.readline())
queen=[]
for q in range(N):
    queen.append(q)
A=[]
ans=0
def Q():
    global ans
    if len(A)==N:
        ans+=1
        return       
    for i in [x for x in queen if x not in A]:
        fac=True
        for j in range(len(A)):
            if abs(j-len(A))==abs(i-A[j]):
                fac=False
                break
        if fac:
            A.append(i)
            Q()
            A.pop()
Q()
print(ans)

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

문제

문제

출처

ICPC

N queens 시간초과 질문드립니다!!

alscks0614 1년 전 1

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말