用Python解N皇后問題

看了一下午的《Python簡明教程》，順手寫了個N皇后的程式玩玩兒。程式只用到了基本的功能。貼在這裡記錄一下。 

def isOK(row,col):
    '''Is it possible that a queen is put at (row,col)?'''
    for i in range(1,row):
        if col==queen[i-1] or col-queen[i-1]==row-i or col-queen[i-1]==i-row:
            return False
    return True
def DFS(row):
    if row==n+1:
        print queen
        return
    for i in range(1,n+1):
        if isOK(row,i):
            queen.append(i)
            DFS(row+1)
            queen.pop()
n=int(raw_input('Enter the number of queens: '))
queen=[]
DFS(1)