回溯

【LeetCode】 51.N 皇后

问题描述

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

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

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

示例 1：
输入：n = 4
输出：[[".Q…","…Q",“Q…”,"…Q."],["…Q.",“Q…”,"…Q",".Q…"]]
解释：如上图所示，4 皇后问题存在两个不同的解法。

示例 2：
输入：n = 1
输出：[[“Q”]]

思路

• 套用回溯算法的模板
• 从第一行开始，先把Q放在第一个位置，（回溯. ，然后放第二个位置、第n个位置）； 然后第二行，检查valid后，放Q（同样回溯放.，再找下一个Q）… 直到第n行放完后， 此时row == n，存放结果。
• valid函数，检查列，两个对角线上是否有Q。

void backtracking(路径, 选择列表)
{
	if(终止条件)
	{
		存放结果;
		return;
	}
	
	for(选择: 选择列表)
	{
		做选择;
		backtracking(路径, 选择列表)
		撤销选择;
	}
}

代码

class Solution {
public:
    vector<vector<string>> solveNQueens(int n) {
        vector<string> board(n, string(n, '.'));
        backtracking(n, 0, board);
        return res;
    }

    void backtracking(int n, int row, vector<string>& board)
    {
        if(row == n)
        {
            res.push_back(board);
            return;
        }

        for(int col =0; col < n; col++)
        {
            if(isVaild(n, row, col, board))
            {
                board[row][col] = 'Q';
                backtracking(n, row + 1, board);
                board[row][col] = '.';
            }
        }
    }

    bool isVaild(int n, int row, int col, vector<string>& board)
    {
        //检查90度
        for(int i = 0; i < row; i++)
        {
            if(board[i][col] == 'Q')
                return false;
        }

        //检查45度
        for(int i = row - 1, j = col + 1; i >= 0 and j < n; i--, j++)
        {   
            if(board[i][j] == 'Q')
                return false;
        }

        //检查135度
        for(int i = row - 1, j = col -1; i >= 0 and j >= 0; i--, j--)
        {   
            if(board[i][j] == 'Q')
                return false;
        }
        return true;
    }
private:
    vector<vector<string>> res;
};

Input: 4
Output: [[".Q..","...Q","Q...","..Q."],["..Q.","Q...","...Q",".Q.."]]

Accepted
9/9 cases passed (4 ms)
Your runtime beats 94.4 % of cpp submissions
Your memory usage beats 77.69 % of cpp submissions (7.2 MB)

思考

把backtracking函数改写成返回bool值，当if(backtracking(n, row + 1, board))我们就返回，这个时候获得一种正确的可能就退出。

    bool backtracking(int n, int row, vector<string>& board)
    {
        if(row == n)
        {
            res.push_back(board);
            return true;
        }

        for(int col =0; col < n; col++)
        {
            if(isVaild(n, row, col, board))
            {
                board[row][col] = 'Q';
                if(backtracking(n, row + 1, board))
                    return true;
                board[row][col] = '.';
            }
        }
        return false;
    }

Input: 4
OutPut: [[".Q..","...Q","Q...","..Q."]]