回溯法-排列树 解决N后问题

# coding=utf-8


# 约束函数 判断lst中的点是否在同一斜线上, 如果在,返回False
# lst是一个列表, 用序号index作为横坐标, 对应的值lst[index]作为纵坐标
# 例如 [1, 3]代表点集(0, 1) (1, 3)
def constraint(lst):
    length = len(lst)
    for i in range(length-1):
        for j in range(i+1, length):
            if abs(i-j) == abs(lst[i]-lst[j]):
                return False
    return True


# 回溯函数
# n 棋盘的大小 n*n
# nums 表示纵坐标可选的值
# lst 表示已经选择的皇后的坐标
def backtracing(n, nums, lst):
    if len(nums) == 0:  # 当纵坐标没有值时, 输出
        for x, y in enumerate(lst):
            print '(%d, %d)' % (x, y),
        print
    else:
        for i in nums:  # 不断尝试可能的纵坐标
            temo_lst = [x for x in lst]
            temo_lst.append(i)
            if constraint(temo_lst):  # 满足约束的选取下一个坐标
                temp_nums = [e for e in nums if e not in temo_lst]
                backtracing(n, temp_nums, temo_lst)


if __name__ == '__main__':
    # N = int(raw_input())
    N = 4
    Nums = [I for I in range(N)]
    backtracing(N, Nums, [])