231017刷题记录_N皇后、数独

231017刷题记录
参考代码随想录来刷的
关键词：回溯、N皇后、数独

1 51. N皇后

力扣题目链接
按照国际象棋的规则，皇后可以攻击与之处在同一行或同一列或同一斜线上的棋子。

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

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

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

该题也是回溯问题。
以示例1为例，
要从第1行中选一个列标，从第2行中选一个列标，从第3行中选一个列标，从第4行中选一个列标。
每次确定列标之前，要判断该坐标是否与已存在的冲突。如果冲突，则要跳过该坐标。

使用path来存列坐标，会简化判断冲突的操作，但是这个解法中将path转化为棋盘的步骤没有处理好，不知道该怎么简化。

class Solution {
    private List<List<String>> result;
    private List<Integer> path;
    public List<List<String>> solveNQueens(int n) {
        result = new ArrayList<>();
        path = new ArrayList<>();
        backtracking(n);
        return result;
    }
    private void backtracking(int n) {
        if (path.size() >= n) {
            List<String> temp = new ArrayList<>();
            for (Integer integer : path) {
                StringBuilder sb = new StringBuilder();
                for (int j = 0; j < integer; j++) {
                    sb.append(".");
                }
                sb.append("Q");
                for (int j = integer + 1; j < n; j++) {
                    sb.append(".");
                }
                temp.add(sb.toString());
            }
            result.add(temp);
            return;
        }
        for (int i = 0; i < n; i++) {
            // (row, i)代表当前坐标，要判断当前坐标是否与已存在的皇后冲突
            int row = path.size();
            boolean conflict = false;
            for (int j = 0; j < path.size(); j++) {
                // (j, queenCol) 目前path中的皇后坐标
                int queenCol = path.get(j);
                if (queenCol == i || i - row == queenCol - j || j + queenCol == row + i) {
                    conflict = true;
                    break;
                }
            }
            if (conflict) continue;
            path.add(i);
            backtracking(n);
            path.remove(path.size() - 1);
        }
    }
}

2 52. N皇后 II

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

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

该题和上面那道题类似。

public class Solution {
    private int result;
    private List<Integer> path;
    public int totalNQueens(int n) {
        result = 0;
        path = new ArrayList<>();
        backtracking(n);
        return result;
    }
    private void backtracking(int n) {
        if (path.size() >= n) {
            result++;
            return;
        }
        for (int i = 0; i < n; i++) {
            // (curRow, i) 表示摆放皇后的坐标
            // (j, queenCol) 表示棋盘上已经摆放的皇后坐标
            int curRow = path.size();
            boolean conflict = false;
            for (int j = 0; j < curRow; j++) {
                int queenCol = path.get(j);
                if (i == queenCol || i - curRow == queenCol - j || curRow + i == j + queenCol) {
                    conflict = true;
                    break;
                }
            }
            if (conflict) continue;
            path.add(i);
            backtracking(n);
            path.remove(path.size() - 1);
        }
    }
}

3 37. 解数独

力扣题目链接
编写一个程序，通过填充空格来解决数独问题。

数独的解法需 遵循如下规则：

1. 数字 1-9 在每一行只能出现一次。
2. 数字 1-9 在每一列只能出现一次。
3. 数字 1-9 在每一个以粗实线分隔的 3x3 宫内只能出现一次。（请参考示例图）

数独部分空格内已填入了数字，空白格用 ‘.’ 表示。

public class Solution {
    public void solveSudoku(char[][] board) {
        int[] first = findFirst(board);
        if (first == null)
            return;
        backtracking(board, first[0], first[1]);
    }
    private boolean backtracking(char[][] board, int row, int col) {
        for (int i = 1; i < 10; i++) {
            if (!isValid(board, row, col, (char)(i + '0'))) continue;
            board[row][col] = (char)(i + '0');
            int[] next = findNext(row, col, board);
            if (next == null) {
                return true;
            }
            if (backtracking(board, next[0], next[1])) return true;
            board[row][col] = '.';
        }
        return false;
    }
    private int[] findNext(int row, int col, char[][] board) {
        for (int i = row; i < 9; i++) {
            for (int j = col + 1; j < 9; j++) {
                if (board[i][j] == '.')
                    return new int[]{i, j};
            }
            col = -1;
        }
        return null;
    }
    private int[] findFirst(char[][] board) {
        for (int i = 0; i < 9; i++) {
            for (int j = 0; j < 9; j++) {
                if (board[i][j] == '.')
                    return new int[]{i, j};
            }
        }
        return null;
    }
    private boolean isValid(char[][] board, int row, int col, char val) {
        // 行
        for (int i = 0; i < 9; i++) {
            if (board[row][i] == val)
                return false;
        }
        // 列
        for (int i = 0; i < 9; i++) {
            if (board[i][col] == val)
                return false;
        }
        // 3x3
        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;
    }
}