代码随想录day30 回溯算法最终章

51. N皇后

题目

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

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

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

示例 1:

代码随想录day30 回溯算法最终章_第1张图片

  • 输入:n = 4
  • 输出:[[".Q..","...Q","Q...","..Q."],["..Q.","Q...","...Q",".Q.."]]
  • 解释:如上图所示,4 皇后问题存在两个不同的解法。

思考

看到题就懵了,感觉和之前做的回溯算法完全不一样,这是真的开放性的题,开了卡哥的视频大概梳理出来了一些思路:

1、要创建的结果数组是三维的,因为棋盘是二维的,要存放棋盘

2、因为棋盘是n*n的,那么当纵向递归遍历到n时,收获结果

3、单层递归其实没啥难度,主要是判断当满足题意三个条件时才能纵向递归,注意这里要因为是在二维数组里递归,那么横向遍历的i和纵向递归遍历其实没啥关系,不用startIndex,直接row+1即可

4、满足的三个条件怎么写:

  • 不能同一列出现两个Q,即col不变,判断从0到row的所有数是否有Q
  • 不能45度出现Q,即3,3到2,2到1,1或者3,2到2,1不能出出现Q,row-1,col-1即可
  • 不能135度出现Q,出3,3到2,4到1,5不能出现Q,row-1,col+1即可

代码

class Solution {

public:

    vector> res;

    void backTracking(vector& chess, int n, int row){

        if(row == n) {

            res.push_back(chess);

            return;

        }

        for(int col = 0; col < n; col++) {

            if(isValid(row, col, chess, n)){

                chess[row][col] = 'Q';

                backTracking(chess, n, row+1);

                chess[row][col] = '.';

            }

        }

    }

    bool isValid(int row, int col, vector chess, int n ) {

        for(int i = 0; i < row; i++) {

            if(chess[i][col] == 'Q') return false;

        }

        for(int i = row-1, j = col-1; i>=0 && j>=0; i--&j--) {

            if(chess[i][j]== 'Q') return false;

        }

        for(int i = row -1, j = col+1; i>=0 && j

            if(chess[i][j] == 'Q') return false;

        }

        return true;

    }

    vector> solveNQueens(int n) {

        vector chess(n, string(n, '.'));

        backTracking(chess, n, 0);

        return res;

    }

};

37. 解数独

题目

编写一个程序,通过填充空格来解决数独问题。

一个数独的解法需遵循如下规则: 数字 1-9 在每一行只能出现一次。 数字 1-9 在每一列只能出现一次。 数字 1-9 在每一个以粗实线分隔的 3x3 宫内只能出现一次。 空白格用 '.' 表示。

代码随想录day30 回溯算法最终章_第2张图片

一个数独。

代码随想录day30 回溯算法最终章_第3张图片

答案被标成红色。

提示:

  • 给定的数独序列只包含数字 1-9 和字符 '.' 。
  • 你可以假设给定的数独只有唯一解。
  • 给定数独永远是 9x9 形式的。

思考

太难了,感觉和n皇后是差不多的,也是三个条件,但是具体应该怎么写完全一头雾水,看完卡哥视频稍微总结一下吧:

  1. 上来不是void函数,是bool类型的,直接遍历,i<9,j<9
  2. 当board[i][j] =='.'时,说明需要填充数字,开始循环判断1-9中哪个数字满足条件,如果满足,直接填上
  3. 注意,这里的中止条件竟然在填充完数字后,如果填充了,那么判断整个函数是否为true,如果为true,返回true
  4. 如果1-9中数字不满足条件,那么要return false
  5. 三个条件写法
  • 纵向判断要填充的数字是否已经在那一列里
  • 横向判断要填充的数字是否已经在那一行里
  • 在3*3的九宫格里判断要填充的数字是否在九宫格里,用两个for循环,并且startRow和StartCol需要用(row/3)*3来表示

代码

class Solution {

private:

    bool backTracking(vectorchar>>& board) {

        for(int i = 0; i < board.size(); i++) {

            for(int j = 0; j < board.size(); j++) {

                if(board[i][j] == '.') {

                    for(char k = '1'; k <= '9'; k++) {

                        if(isValid(i, j, k, board)) {

                            board[i][j] = k;

                            bool tmp = backTracking(board);

                            if(tmp) return true;

                            board[i][j] = '.';

                        }

                    }

                    return false;

                }

            }

        }

        return true;

    }

    bool isValid(int row, int col, char val, vectorchar>>& board) {

        for(int i = 0; i < 9; i++) {

            if(board[row][i] == val) return false;

        }

        for(int j = 0; j < 9; j++) {

            if(board[j][col] == val) return false;

        }

        int startRow = (row/3) * 3;

        int startCol = (col/3) * 3;

        for(int i = startRow; i < startRow+3; i++) {

            for(int j = startCol; j < startCol+3; j++) {

                if(board[i][j] == val) return false;

            }

        }

        return true;

    }

public:

    void solveSudoku(vectorchar>>& board) {

        backTracking(board);

    }

};

你可能感兴趣的:(算法,模拟退火算法)