经典算法题型（三）：N皇后问题

一、基本概念

1.N皇后问题是由国际象棋8皇后问题引申过来的，在国际象棋中有8行8列的棋盘，其中皇后的前进路线为：横、纵、以及双斜线，为了使皇后之间互不冲突，就要摆放皇后，使之前进路线不重叠。

如下所示就是一个8皇后问题的解决方案，如图可以验证，各个皇后之间互不冲突：

Q  .    .    .    .    .    .    .
.    .    .    .   Q   .    .    .
.    .    .    .    .    .    .   Q
.    .    .    .    .   Q   .    .
.    .   Q   .    .    .    .    .
.    .    .    .    .    .   Q   .
.    Q   .   .    .    .    .    . 
.     .   .   Q   .    .    .    .

 2.由此引申到N皇后问题，借用leetcode 51的问题解决该问题

二、问题解析：

/*51. N皇后
n 皇后问题研究的是如何将 n 个皇后放置在 n×n 的棋盘上，并且使皇后彼此之间不能相互攻击。

上图为 8 皇后问题的一种解法。

给定一个整数 n，返回所有不同的 n 皇后问题的解决方案。

每一种解法包含一个明确的 n 皇后问题的棋子放置方案，该方案中 'Q' 和 '.' 分别代表了皇后和空位。

示例:

输入: 4
输出: [
 [".Q..",  // 解法 1
  "...Q",
  "Q...",
  "..Q."],

 ["..Q.",  // 解法 2
  "Q...",
  "...Q",
  ".Q.."]
]
解释: 4 皇后问题存在两个不同的解法。
*
* */

分析

逐行摆放皇后的过程中的，如何判断某一个位置不能再摆放位置？

1.使用一维数组，标记某一行、某一列是否被占用

private boolean[] col;

如果有皇后摆放在第一行或者第一列，则col[1]==true

2.标记两个对角线，这里涉及到两个小技巧

（1）右上：同一对角线相加为同一个数

（2）左下：同一对角线相加为同一个数

为了使其数组下标从0开始，需要在加上n-1;

 

 

三、代码及注释

import java.util.Arrays;
import java.util.LinkedList;
import java.util.List;
import java.util.ArrayList;

public class Solution {

    private boolean[] col;
    private boolean[] dia1;
    private boolean[] dia2;
    private ArrayList<List<String>> res;

    public List<List<String>> solveNQueens(int n) {

        res = new ArrayList<List<String>>();
        col = new boolean[n];
        dia1 = new boolean[2 * n - 1];
        dia2 = new boolean[2 * n - 1];

        LinkedList<Integer> row = new LinkedList<Integer>();
        putQueen(n, 0, row);

        return res;
    }

    // 尝试在一个n皇后问题中, 摆放第index行的皇后位置
    private void putQueen(int n, int index, LinkedList<Integer> row){

        if(index == n){
            res.add(generateBoard(n, row));
            return;
        }

        for(int i = 0 ; i < n ; i ++)
            // 尝试将第index行的皇后摆放在第i列
            if(!col[i] && !dia1[index + i] && !dia2[index - i + n - 1]){
                row.addLast(i);
                col[i] = true;
                dia1[index + i] = true;
                dia2[index - i + n - 1] = true;
                putQueen(n, index + 1, row);
                col[i] = false;
                dia1[index + i] = false;
                dia2[index - i + n - 1] = false;
                row.removeLast();
            }

        return;
    }

    private List<String> generateBoard(int n, LinkedList<Integer> row){

        assert row.size() == n;

        ArrayList<String> board = new ArrayList<String>();
        for(int i = 0 ; i < n ; i ++){
            char[] charArray = new char[n];
            Arrays.fill(charArray, '.');
            charArray[row.get(i)] = 'Q';
            board.add(new String(charArray));
        }
        return board;
    }

    private static void printBoard(List<String> board){
        for(String s: board)
            System.out.println(s);
        System.out.println();
    }

    public static void main(String[] args) {

        int n = 4;
        List<List<String>> res = (new Solution()).solveNQueens(n);
        for(List<String> board: res)
            printBoard(board);
    }
}