算法训练day30-回溯-N皇后

332.重新安排行程

给你一份航线列表 tickets ，其中 tickets[i] = [fromi, toi] 表示飞机出发和降落的机场地点。请你对该行程进行重新规划排序。

所有这些机票都属于一个从 JFK（肯尼迪国际机场）出发的先生，所以该行程必须从 JFK 开始。如果存在多种有效的行程，请你按字典排序返回最小的行程组合。

• 例如，行程 ["JFK", "LGA"] 与 ["JFK", "LGB"] 相比就更小，排序更靠前。

假定所有机票至少存在一种合理的行程。且所有的机票 必须都用一次 且 只能用一次。

思路

因为题目要求返回的组合必须是按照字典排序最小的，所以可以提前对数组元素的目的地进行排序，这样返回的第一个解就是要寻找的组合。

终止条件：当path中的元素数量等于元素长度+1时说明直接找到了解。

需要一个used数组记录当前元素是否已经使用过。

代码

class Solution {
    List<String>  res = new ArrayList<>();
    LinkedList<String> path = new LinkedList<>();
    public List<String> findItinerary(List<List<String>> tickets) {
        // 对到达地点排序
        Collections.sort(tickets, (a, b) -> a.get(1).compareTo(b.get(1)));
        path.add("JFK");
        // 定义uesed数组，用来记录元素是否已经使用过
        boolean[] used = new boolean[tickets.size()];
        backtracking(tickets,used);
        return res;  
    }
    private boolean backtracking(List<List<String>> tickets,boolean[] used){
        // 终止条件：当path中元素数量等于tickets的长度+1时，返回结果
        if(path.size()==tickets.size()+1){
            res = new ArrayList(path);
            return true;
        }
        for(int i=0;i<tickets.size();i++){
            // 当前地点没用过且当前地点的起始地址与上个目的地相同
            if(!used[i]&&tickets.get(i).get(0).equals(path.getLast())){
                // path 中加入的是当前节点的目的地址
                path.add(tickets.get(i).get(1));
                used[i]  = true;
                if(backtracking(tickets,used)){
                    return true;
                }
                used[i] = false;
                path.removeLast();
            }
        }
        return false;
    } 

}

51.N皇后

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

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

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

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

思路

终止条件：当搜索到达叶子节点也就是最后一行时，收集并返回结果。

另外需要写一个函数检查当前行列，斜线上是否已经有皇后了，如果有返回false。

代码

class Solution {
    List<List<String>>  res = new ArrayList<>();
    public List<List<String>> solveNQueens(int n) {
       // 创建一个棋盘数组
       char[][] chessboard = new char[n][n];
       for(char[] c :chessboard){
           Arrays.fill(c,'.');
       }
       backtracking(chessboard,0,n);
       return res;
    }

    private void backtracking(char[][] chessboard,int row,int n ){
        // 终止条件：当搜索到叶子节点即搜索到最后一行时，采集结果并返回
        if(row==n){
            res.add(Array2List(chessboard));
            return;
        }
        for(int col=0;col<n;col++){
            if(isValid(row,col,n,chessboard)){
                chessboard[row][col] = 'Q';
                backtracking(chessboard,row+1,n);
                chessboard[row][col] = '.';
            }
        }
    }

    private List Array2List(char[][] chessboard) {
        List<String> list = new ArrayList<>();

        for (char[] c : chessboard) {
            list.add(String.copyValueOf(c));
        }
        return list;
    }
    
    private boolean isValid(int row, int col, int n, char[][] chessboard) {
        // 检查列
        for (int i=0; i<row; ++i) { // 相当于剪枝
            if (chessboard[i][col] == 'Q') {
                return false;
            }
        }

        // 检查45度对角线
        for (int i=row-1, j=col-1; i>=0 && j>=0; i--, j--) {
            if (chessboard[i][j] == 'Q') {
                return false;
            }
        }

        // 检查135度对角线
        for (int i=row-1, j=col+1; i>=0 && j<=n-1; i--, j++) {
            if (chessboard[i][j] == 'Q') {
                return false;
            }
        }
        return true;
    }
}

37.解数独

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

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

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

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

思路

首先本题与之前的题目不同，本题是在一个二维数组的遍历，所以需要从行和列上同时遍历。

当行和列遍历到最后一个元素时，程序自然也就终止了。当要注意的是每找到一个合适的组合需要立即填充到数组中，所以当找到一个满足的结果就要返回。

代码

class Solution {
    public void solveSudoku(char[][] board) {
       backtracking(board);
       return ;
    }
    private boolean backtracking(char[][] board){
       // 遍历行
       for(int i=0;i<board.length;i++){
        // 遍历列
        for(int j=0;j<board[0].length;j++){
            if(board[i][j]!='.') continue;
            // [1,j]这个位置依次放1->9尝试
            for(char k='1';k<='9';k++){
                if(isValid(i,j,k,board)){
                    board[i][j] = k;
                    // 如果找到合适的一组立即返回
                    if(backtracking(board)) return true;
                    // 回溯
                    board[i][j] = '.';
                }
            }
            // 9个数遍历完没找到返回false
            return false;
        }
       }
        // 遍历完返回true
       return true;
    }

    /**
     * 判断棋盘是否合法
     *  同行是否重复
     *  同列是否重复
     *  9宫格是否重复
     */
     private boolean isValid(int row, int col, char val, char[][] 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;
            }
        }
        // 9宫格里是否重复
        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;
     }
}