LeetCode51. N-Queens N皇后问题

The n-queens puzzle is the problem of placing n queens on an n×n chessboard such that no two queens attack each other.

八皇后

Given an integer n, return all distinct solutions to the n-queens puzzle.

Each solution contains a distinct board configuration of the n-queens' placement, where 'Q' and '.' both indicate a queen and an empty space respectively.

N皇后问题是一个经典的算法问题,求N×N的国际象棋棋盘上能摆放多少个皇后,且她们之间不相互攻击,要输出所有可能的摆法。皇后的攻击范围是同一行、同一列以及同在两条45°斜线上的棋子。

解本题的关键是选择一个合适的数据结构来记录放皇后的位置,还要方便检验各个位置是否冲突。使用x和y坐标来记录位置是最直接的,然而保存两个数要用额外的数据结构才行(特殊情况可以用一个int打包),记录下来后判断位置是否冲突也比较麻烦。

经过思考,使用一个int[N]数组来保存皇后的位置是最合适的。数组下标代表y坐标,数组中的数0~N-1代表x坐标。这样保证y坐标不会重复(不会在同一行),判断x坐标是否重复只要遍历数组看是否有重复数字,判断是否在同一斜线也很方便:假设当前位置是int[yi]=xi,从当前位置向前移动,判断 (int[yi-1]==xi+1 || int[yi-1]==xi-1), (int[yi-12]==xi+2 || int[yi-2]==xi-2)...即可。

解决了数据结构和位置检查的问题,剩下的算法就很简单:从位置0开始,检查当前位置能放0~N-1中的哪个数,如果能放且是最后一个数,就把数组转换一下输出,如果没到最后一个数,就对下一个位置递归计算。

代码如下,本算法击败了98%的java算法。

    public static List<List<String>> solveNQueens(int n) {
        List<List<String>> res = new ArrayList<>();
        int[] a = new int[n];
        solveNQueens(res, a, 0);
        return res;
    }

    public static void solveNQueens(List<List<String>> res, int[] a, int pos) {
        int n = a.length;
        for(int j = 0; j < n; j++) {
            if(check(a, pos, j)) {
                a[pos] = j;
                if(pos == n - 1) { // save
                    char[][] board = new char[n][n];
                    for(char[] b : board) {
                        Arrays.fill(b, '.');
                    }
                    List<String> list = new ArrayList<>();
                    for(int i = 0; i < n; i++) {
                        board[i][a[i]] = 'Q';
                        list.add(new String(board[i]));
                    }
                    res.add(list);
                } else solveNQueens(res, a, pos + 1);
            }
        }
    }

    public static boolean check(int[] a, int pos, int j) {
        for(int k = pos - 1; k >= 0; k--) {
            if(a[k] == j) return false;
        }
        for(int k = pos - 1; k >= 0; k--) {
            int diff = pos - k;
            if(a[k] == j + diff || a[k] == j - diff) return false;
        }
        return true;
    }
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 214,377评论 6 496
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 91,390评论 3 389
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 159,967评论 0 349
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 57,344评论 1 288
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 66,441评论 6 386
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 50,492评论 1 292
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,497评论 3 412
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,274评论 0 269
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,732评论 1 307
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,008评论 2 328
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,184评论 1 342
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,837评论 4 337
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,520评论 3 322
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,156评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,407评论 1 268
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,056评论 2 365
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,074评论 2 352

推荐阅读更多精彩内容

  • 回溯算法 回溯法:也称为试探法,它并不考虑问题规模的大小,而是从问题的最明显的最小规模开始逐步求解出可能的答案,并...
    fredal阅读 13,643评论 0 89
  • 贪心算法 贪心算法总是作出在当前看来最好的选择。也就是说贪心算法并不从整体最优考虑,它所作出的选择只是在某种意义上...
    fredal阅读 9,225评论 3 52
  • 原文地址:http://theory.stanford.edu/~amitp/GameProgramming/ 1...
    达微阅读 19,432评论 0 28
  • 走上工作岗位十五年,参加的培训大大小小的不算多,却也不少。最长的一次培训也只有一周,培训方式往往是以面授形...
    陕县2930孙敏阅读 59评论 0 0
  • 夜深人静一棵树在长在风中它晃动着,同时也在长它动一下长一点再动一下又长一点 在一棵树的体内藏满了那些生长的细胞如同...
    王春涞阅读 153评论 7 5