글 읽기 - 9프로에서 시간초과가 나요

waroad 3년 전 0

어떻게 하면 더 빠르게 할 수 있을까요?

아무리 해도 시간초과가 나네요.

S=int(input())
arr = [[int(y) for y in input().split()] for x in range(S)]
cnt=0
ans=0
queen=[[],[]]


def check(a,b,c):
    for i in range(0,len(queen[c])):
        if abs(queen[c][i][0]-a)==abs(queen[c][i][1]-b):
            return 0
    return 1


def do(a,b):
    global cnt, arr, ans
    for i in range(a,S):
        for j in range(0,S):
            if i==a and j<b:
                continue
            if arr[i][j]==1:
                c = (i + j) % 2
                if check(i, j, c)==0: continue
                queen[c].append([i,j])
                cnt+=1
                if ans<cnt:
                    ans=cnt
                do(i,j)
                cnt-=1
                queen[c].pop()


do(0,0)
'''for i in range(0,S):
    print(*arr[i])'''
print(ans)

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

문제

문제

출처

ICPC

9프로에서 시간초과가 나요

waroad 3년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말