代码随想录算法训练营补打卡 day30| 332.重新安排行程、51. N皇后、37. 解数独

代码随想录

332.重新安排行程

思路

题解思路：
首先给数组排好序，按照升序，然后一旦找到返回即可！！！
string的conpareTo方法是重写好的。注意一定要升序，然后给回溯函数一个返回值，这样一旦找到了第一个返回值就返回，不用遍历整个回溯过程！！！！

代码

class Solution {
     List<String> results = new ArrayList<String>();
     List<String> res = new ArrayList<String>();
    public List<String> findItinerary(List<List<String>> tickets) {
        Collections.sort(tickets, (a, b) -> a.get(1).compareTo(b.get(1)));
        Boolean used[] = new Boolean[tickets.size()];
        Arrays.fill(used,false);
        results.add("JFK");
        backtracking(tickets,used);
        return res;
    }

    public Boolean backtracking(List<List<String>> tickets,Boolean used[]){
        if(results.size() == tickets.size()+1){
            res = new ArrayList<String>(results);
            return true;
        }

        for(int i = 0;i<tickets.size();i++){
            
            if(!(tickets.get(i).get(0)).equals(results.get(results.size()-1))) continue;
            if(used[i]) continue;//树枝去重

            
            results.add(tickets.get(i).get(1));
            used[i] = true;
            if(backtracking(tickets,used))
                return true;
            used[i] = false;
            results.remove(results.size()-1);
        }
        return false;

    }

}

51. N皇后

思路

题解思路：用char数组来保存棋盘。
棋盘的宽度即为遍历的次数，深度即为递归的次数，终值条件就是到达最后一行。
每次
利用String.copyValueOf来复制char数组到string中。
每次判断是否满足不会攻击的条件

代码

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, '.');
        }
        backTrack(n, 0, chessboard);
        return res;
    }

    public void backTrack(int n, int row, char[][] chessboard) {
        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';
                backTrack(n, row+1, chessboard);
                chessboard[row][col] = '.';
            }
        }

    }

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

        for (char[] c : chessboard) {
            list.add(String.copyValueOf(c));
        }
        return list;
    }

    public 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. 解数独

思路

题解思路：二维递归，因为它需要每个位置都遍历过去，所以需要对行和列都做一个递归。
当得到一个正确的解就返回，否则返回false

代码

class Solution {
    public void solveSudoku(char[][] board) {
        backtracking(board);
    }

    public Boolean backtracking(char[][] board){

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

                if(board[i][j]!='.') continue;
                for(char k = '1';k<='9';k++){
                    if(isValidSudoku(i,j,k,board)){
                        board[i][j]=k;
                        if(backtracking(board)) return true;
                        board[i][j]='.';
                    }
                }
                return false;

                
                
            }
        }

        return true;

    }

     /**
     * 判断棋盘是否合法有如下三个维度:
     *     同行是否重复
     *     同列是否重复
     *     9宫格里是否重复
     */
    private boolean isValidSudoku(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;
    }
}