代码随想三刷回溯篇4

332. 重新安排行程

题目

链接

代码

一个超过时间限制

class Solution {
    public List<String> findItinerary(List<List<String>> tickets) {
        Collections.sort(tickets,(e1,e2)->e1.get(1).compareTo(e2.get(1)));
        paths.add("JFK");
        used = new boolean[tickets.size()];
        backtracking(tickets);
        return result;
    }
    List<String> paths = new ArrayList();
    List<String> result = null;
    boolean[] used;

    public void backtracking(List<List<String>> tickets){
        if(result!=null){
            return;
        }
        if(paths.size()==tickets.size()+1){
            result = new ArrayList(paths);
            return;
        }
        for(int i=0;i<tickets.size();i++){
            if(!used[i]&&tickets.get(i).get(0).equals(paths.get(paths.size()-1))){
                paths.add(tickets.get(i).get(1));
                used[i] = true;
                backtracking(tickets);
                used[i] = false;
                paths.remove(paths.size()-1);
            }
        }
    }
}

51. N 皇后

题目

链接

代码

class Solution {
    public List<List<String>> solveNQueens(int n) {
        if(n==1){
            List<String> paths= new ArrayList();
            paths.add("Q");
            result.add(paths);
            return result;
        }
        if(n==0||n<=3){
            return result;
        }
        char[][] arr = new char[n][n];
        for (char[] c : arr) {
            Arrays.fill(c, '.');
        }
        backtracking(arr,0);
        return result;
    }
    List<List<String>> result = new ArrayList();
    int count = 0;
    public void backtracking(char[][] arr,int i){
        if(i==arr.length&&count==arr.length){
            List<String> paths= new ArrayList();
            for(int k = 0;k<arr.length;k++){
                StringBuilder sb = new StringBuilder();
                for(int j = 0;j<arr.length;j++){
                    sb.append(arr[k][j]);
                }
                paths.add(sb.toString());
            }
            result.add(paths);
            return;
        }
        if(i>=arr.length||count!=i){
            return;
        }
        for(int j =0;j<arr.length;j++){
            if(checkDown(arr,i,j)){
                arr[i][j] = 'Q';
                count++;
                backtracking(arr,i+1);
                count--;
                arr[i][j] = '.';
            }
        }
    }

    public boolean checkDown(char[][] arr,int i,int j){
        
        int tempi = i;
        int tempj = j;
        while(i!=0&&j!=0){
            if(arr[i-1][j-1] =='Q'){
                return false;
            }
            j--;
            i--;
        }
        i = tempi;
        j = tempj;
        while(i!=0){
            if(arr[i-1][j] =='Q'){
                return false;
            }
            i--;
        }
        i = tempi;
        j = tempj;
        while(i!=0&&j<arr.length-1){
            if(arr[i-1][j+1] =='Q'){
                return false;
            }
            i--;
            j++;
        }
        

        return true;
    }
}

37. 解数独

题目

链接

代码

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

    private boolean solveSudokuHelper(char[][] board){
        //「一个for循环遍历棋盘的行，一个for循环遍历棋盘的列，
        // 一行一列确定下来之后，递归遍历这个位置放9个数字的可能性！」
        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++){ // (i, j) 这个位置放k是否合适
                    if (isValidSudoku(i, j, k, board)){
                        board[i][j] = k;
                        if (solveSudokuHelper(board)){ // 如果找到合适一组立刻返回
                            return true;
                        }
                        board[i][j] = '.';
                    }
                }
                // 9个数都试完了，都不行，那么就返回false
                return false;
                // 因为如果一行一列确定下来了，这里尝试了9个数都不行，说明这个棋盘找不到解决数独问题的解！
                // 那么会直接返回， 「这也就是为什么没有终止条件也不会永远填不满棋盘而无限递归下去！」
            }
        }
        // 遍历完没有返回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;
    }
}