八皇后问题 Python 版

题目
给定一个 N*N 正方形棋盘,在上面放置 N个棋子,又叫皇后,使每两个棋子都不在同一条横线上、竖线上、斜线上。一般我们都讨论8皇后,但是只要N > 4,都会存在解的。
分析
方法1:根据定义来处理,即每往棋盘中放置皇后的时候,都要判断哪些位置可以放新加入的皇后,而哪些地方如果放置皇后的话,会造成冲突。我下面写的这个代码就是基于此。
方法2、我看了下别人的优化,主要是采用位运算来实现计算复杂度降低的,我没有用Python 实现这个,所以在这里挖一个坑
代码
代码里的注释有详细的说明,设定N值,即可返回一个符合要求的解。但是这个问题还有一个进阶,那就是讨论究竟会有多少个解,这就需要数论的知识了,而且我对这块数学没研究过,所以也没有代码实现。大家将就使用一下代码就好了。
class EightQueensPuzzle(object):
    '''
    八皇后问题求解
    代码使用方法:
        eight_q = EightQueensPuzzle(4, 5)
        print "EIGHT QUEEDS PUZZLE:"
        result = eight_q.eight_queens_puzzle()
        for i in result:
            print i
    '''
    def __init__(self, n, char):
        self.n = n  # 棋盘维度
        self.char = char  # 皇后标记字符

    def init_chess_board(self, n):
        '''
        初始化一个棋盘,棋盘规格可以按参数 n 随意选定,一般都讨论八皇后,就选择 8
        :return: 返回棋盘,是一个 8*8 矩阵
        '''
        chess_board = []
        for i in xrange(0, n):
            line = []
            for j in xrange(0, n):
                line.append(0)
            chess_board.append(line)
        return chess_board

    def update_conflict_board(self, conflict_board, position):
        for k in xrange(0, self.n):  # 为行添加 1
            conflict_board[position[0]][k] = 1
        for id in xrange(position[0]+1, self.n):
            conflict_board[id][position[1]] = 1  # 为列添加 1
            if position[0] + position[1] - id >= 0:  # 为左斜添加 1
                conflict_board[id][position[0] + position[1] - id] = 1
            if position[1] - position[0] + id < self.n:  # 为右斜添加 1
                conflict_board[id][position[1] - position[0] + id] = 1

    def queens_conflict(self, conflict_board, position):
        '''
        当前棋盘的状态是 conflict_board, 判定如果在 position 位置给一个皇后的话,会不会出现问题。
        如果有问题则返回 False,如果没有问题返回 True
        '''
        if conflict_board[position[0]][position[1]] != 0:
            return False
        else:
            return True

    def eight_queens_puzzle(self):
        '''
        给出一个八皇后的求解答案。
        :return:返回一个结果并打印.
        '''
        import random
        while True:  # 不停寻找符合条件的八皇后排列
            chess_board = self.init_chess_board(self.n)
            conflict_board = self.init_chess_board(self.n)
            for i in xrange(0, self.n):
                flag = 0
                for cnt in conflict_board[i]:
                    if cnt != 0:
                        flag += 1
                if flag == self.n:  # 如果已经1被填满了,说明这个答案错误
                    break

                while True:
                    pos = [i, random.randint(0, self.n-1)]  # 元组构成皇后的位置
                    if self.queens_conflict(conflict_board, pos):  # 如果没有冲突
                        chess_board[i][pos[1]] = self.char
                        self.update_conflict_board(conflict_board, pos)
                        break
            if self.char in chess_board[self.n-1]:
                return chess_board
发布了74 篇原创文章 · 获赞 67 · 访问量 11万+
展开阅读全文

python皇后中的列表问题

02-06

我这里写了一个简单的八皇后程序,用的是最笨的算法,大家就不要吐槽我的算法了,问题是我的保存所有解法的总列表不正常,列表中所有元素都一样,劳烦大家看下问题出在哪 ``` import time import os def check(list_): #这个函数用来检查当前位置是否合法 for i in range(0,len(list_)-1): if len(list_)==1 : continue for j in range(i+1,len(list_)): x=list_[i][0] y=list_[j][0] x1=list_[i][1] y1=list_[j][1] if (x==y or x1==y1 or (x-y)**2==(x1-y1)**2): return False return True def find_position(list_): #这个函数用来寻找下一可尝试位置 if list_ : if list_[-1][1]==num-1: list_.pop() find_position(list_) else: return list_ else: return False def print_(list_,num): #直观的打印摆法 for i in range(num): print() for j in range(num): if list_[i][1]!=j : print('十', end='') else: print('㊣',end='') list_=[(0,0)] #初始第一个解法列表 sumlist=[] #所有解法构成的列表 num=8 #皇后数 column=0 #每行开始时尝试的列号 tag=0 if not os.path.isfile(str(num)+'皇后.txt'): #输出到记事本 f=open(str(num)+'皇后.txt','w+') tag=1 starttime=time.time() while(len(list_) and len(list_)<num+1): #这里八皇后排到第九行,保证前八行摆的合法 if check(list_): list_.append((list_[-1][0]+1,column)) else: find_position(list_) tuple_ =list_[-1] list_.pop() list_.append((tuple_[0],tuple_[1]+1)) list_.pop() #找到一个解 sumlist.append(list_) #去掉第九行 solution=1 #记录总的解法数 print_(list_,num) if tag==1 : print(solution,list_,file=f) while len(list_) : # list_.pop() #退到第七行 find_position(list_) #找到下一可尝试位置 tuple_ =list_[-1] list_.pop() list_.append((tuple_[0],tuple_[1]+1)) while(len(list_) and len(list_)<num+1): if check(list_): list_.append((list_[-1][0]+1,column)) else: find_position(list_) if list_: tuple_ =list_[-1] list_.pop() list_.append((tuple_[0],tuple_[1]+1)) if list_: list_.pop() #print(list_) sumlist.append(list_) solution+=1 if tag==1 : print(solution,list_,file=f) endtime=time.time() #print(sumlist[-2],sumlist[-1]) #去掉这里的注释可以发现最后两个元素一样! if tag==1 : f.close() #大家可以去文件里查看所有解法确实生成了 print(sumlist) #但sumlist所有元素都是最后一个lsit_的值,也就是空,这里就是我的问题 print(solution) #总的解法数 print((endtime-starttime)/solution) #生成每个解法的平均时间 #0.0089,0.0086,0.0143,0.0244 #92, 352, 724, 2680 ``` 这就我的程序,我感觉是我对Python的变量概念理解的不好,希望大神赐教 问答

没有更多推荐了,返回首页

©️2019 CSDN 皮肤主题: 大白 设计师: CSDN官方博客

分享到微信朋友圈

×

扫一扫,手机浏览