算法:回溯题目总结

回溯法解决的问题

回溯法,一般可以解决如下几种问题:

组合问题:N个数里面按一定规则找出k个数的集合
切割问题:一个字符串按一定规则有几种切割方式
子集问题:一个N个数的集合里有多少符合条件的子集
排列问题:N个数按一定规则全排列,有几种排列方式
棋盘问题:N皇后,解数独等等

组合问题

题目1:77. 组合

思路:循环套递归加回溯
1)递归法

class Solution {
    List<List<Integer>> res=new ArrayList<>();
    LinkedList<Integer> path=new LinkedList<>();
    public List<List<Integer>> combine(int n, int k) {
        backtracking(n,k,1);
        return res;
    }
    //n是范围 k是k个数的组合 startIndex是控制取下一个节点继续组合
    public void backtracking(int n,int k,int startIndex){
        //收集结果
        if(path.size()==k){
            //说明拿到了一个个数为k的组合
            res.add(new ArrayList<>(path));
        }
        //for循环控制节点不重复 取到的组合也不重复
        for(int i=startIndex;i<=n;i++){
            path.add(i);
            //递归
            backtracking(n,k,i+1);
            //回溯
            path.removeLast();
        }
    }
}
题目2:216. 组合总和 III

思路:回溯模版变种题
1)递归法

class Solution {
    List<List<Integer>> result=new ArrayList<>();
    LinkedList<Integer> path=new LinkedList<>();

    public List<List<Integer>> combinationSum3(int k, int n) {
        backtracking(k,n,0,1);
        return result;
    }

    public void backtracking(int k,int n,int sum ,int startIndex){
        //如果和为n 并且 结果是k个 收集结果
        if(sum==n&&path.size()==k){
            result.add(new ArrayList<>(path));
        }
        for(int i=startIndex;i<=9;i++){
            path.add(i);
            sum+=i;
            backtracking(k,n,sum,i+1);
            //回溯
            sum-=i;
            path.removeLast();
        }
    }
}

剪枝优化后的代码

class Solution {
    List<List<Integer>> result=new ArrayList<>();
    LinkedList<Integer> path=new LinkedList<>();

    public List<List<Integer>> combinationSum3(int k, int n) {
        backtracking(k,n,0,1);
        return result;
    }

    public void backtracking(int k,int n,int sum ,int startIndex){
    	// 剪枝操作
        if (sum > n) { 
            return;
        }    
        // 剪枝操作 收集的元素比k大 
        if (path.size() > k) return;
        //如果和为n 并且 结果是k个 收集结果
        if(sum==n&&path.size()==k){
            result.add(new ArrayList<>(path));
        }
        for(int i=startIndex;i<=9;i++){
            path.add(i);
            sum+=i;
            backtracking(k,n,sum,i+1);
            //回溯
            sum-=i;
            path.removeLast();
        }
    }
}
*题目3:17. 电话号码的字母组合

思路:得复习 还挺难理解
1)递归法

class Solution {

    List<String> result=new ArrayList<>();
    StringBuilder path=new StringBuilder();  
    public List<String> letterCombinations(String digits) {
        if(digits==null || digits.length()==0){
            return result;
        }
        String[] numString={"","","abc","def","ghi","jkl","mno","pqrs","tuv","wxyz"};
        backtracking(digits,numString,0);
        return result;
    }

    public void backtracking(String digits,String[] numString,int startIndex){
       //收集结果
       if(startIndex==digits.length()){
           result.add(path.toString());
           return;
       }
       //取出数字startIndex对应的字母
        String str=numString[digits.charAt(startIndex)-'0'];
        for(int i=0;i<str.length();i++){
            path.append(str.charAt(i));
            backtracking(digits,numString,startIndex+1);
            path.deleteCharAt(path.length()-1);
        }
    }
}
题目4:39. 组合总和

思路:这个是可以重复选取元素的题 与前两题不同 就是一个递归参数的变化 不加一就可以
1)递归法

class Solution {
    private List<List<Integer>> result=new ArrayList<>();
    private LinkedList<Integer> path=new LinkedList<Integer>();
    public List<List<Integer>> combinationSum(int[] candidates, int target) {
        if(candidates.length==0){
            return result;
        }
        bracktracking(candidates,target,0,0);
        return result;
    }
    public void bracktracking(int[] candidates, int target,int sum,int startIndex){
        if(sum>target){
            return;
        }
        if(sum==target){
            result.add(new ArrayList<>(path));
        }
        for(int i=startIndex;i<candidates.length;i++){
            sum+=candidates[i];
            path.add(candidates[i]);
            //注意 此时i不+1
            bracktracking(candidates,target,sum,i);
            sum-=candidates[i];
            path.removeLast();
        }
    }
}
*题目5:40. 组合总和 II

思路:这题有重复元素需先排序 其次要借助一下used数组来数层剪枝 有难度 需复习
1)递归法

class Solution {
    private List<List<Integer>> result=new ArrayList<>();
    private LinkedList<Integer> path=new LinkedList<Integer>();
    public List<List<Integer>> combinationSum2(int[] candidates, int target) {
         if(candidates.length==0){
            return result;
         }
        //排序
        Arrays.sort(candidates);
        int[] used=new int[candidates.length];
        bracktracking(candidates,target,0,0,used);
        return result;
    }
    public void bracktracking(int[] candidates, int target,int sum,int startIndex,int[] used){
        if(sum>target){
            return;
        }
        if(sum==target){
            result.add(new ArrayList<>(path));
        }
        for(int i=startIndex;i<candidates.length;i++){
        	//去重逻辑
            if(i>0&&candidates[i]==candidates[i-1]&&used[i-1]==0){
                continue;
            }
            sum+=candidates[i];
            path.add(candidates[i]);
            used[i]=1;
            bracktracking(candidates,target,sum,i+1,used);
            sum-=candidates[i];
            path.removeLast();
            used[i]=0;
        }
    }
}

切割问题

*题目6:131. 分割回文串

思路:因为切割过的地方,不能重复切割 和组合的回溯过程差不多 双指针法判断回文串
难点:
切割问题可以抽象为组合问题
如何模拟那些切割线
切割问题中递归如何终止
*在递归循环中如何截取子串
如何判断回文
1)递归法

class Solution {
    List<List<String>> res=new ArrayList<>();
    LinkedList<String> path=new LinkedList<>();
    public List<List<String>> partition(String s) {
        backtracking(s,0);
        return res;
    }
    private void backtracking(String s,int startIndex){
        //如果起始位置大于s的大小,说明找到了一组分割方案
        if (startIndex >= s.length()) {
            res.add(new ArrayList(path));
            return;
        }
        for(int i=startIndex;i<s.length();i++){
            //判断是否是回文串(难点)
            if(checkPalindRome(s,startIndex,i)){
                //切割子串(难点)
                String str = s.substring(startIndex, i+1);
                path.add(str);
            } else {
                continue;
            }
            backtracking(s, i + 1);
            path.removeLast();
        }
    }

    private boolean checkPalindRome(String s, int start, int end){
        for (int i = start, j = end; i < j; i++, j--) {
            if (s.charAt(i) != s.charAt(j)) {
                return false;
            }
        }
        return true;
    }
}
**题目7:93. 复原 IP 地址

思路:
1)递归法

class Solution {
    List<String> res = new ArrayList<>();
    public List<String> restoreIpAddresses(String s) {
        StringBuilder sb = new StringBuilder(s);
        backTracking(sb, 0, 0);
        return res;
    }
    private void backTracking(StringBuilder s, int startIndex, int dotCount){
        if(dotCount==3){
            if(isValid(s, startIndex, s.length() - 1)){
                res.add(s.toString());
            }
            return;
        }
        for(int i = startIndex; i < s.length(); i++){
            if(isValid(s, startIndex, i)){
                //如果字符串合法拼上.
                s.insert(i + 1, '.');
                //递归i+2是因为i+1是.
                backTracking(s, i + 2, dotCount + 1);
                //回溯 把字符串还原
                s.deleteCharAt(i + 1);
            }else{
                break;
            }
        }
    }
    //[start, end]
    private boolean isValid(StringBuilder s, int start, int end){
        if(start > end) return false;
        // 0开头的数字不合法
        if(s.charAt(start) == '0' && start != end) return false;
        //当前字符串拼接的数
        int num = 0;
        for(int i = start; i <= end; i++){
            //当前字符串对应的数字
            int digit = s.charAt(i) - '0';
            num = num * 10 + digit;
            if(num > 255) return false;
        }
        return true;
    }
}

子集问题

组合问题和分割问题收集结果的时候相当于收集的是部分 而子集收集的是全部 所以没有判断==k去收集结果

题目8:78. 子集

思路:
1)递归法

class Solution {
    List<List<Integer>> result=new ArrayList<>();
    LinkedList<Integer> path=new LinkedList<>();
    public List<List<Integer>> subsets(int[] nums) {
        if(nums.length==0){
            return result;
        }
        bracktrack(nums,0);
        return result;
    }
    void bracktrack(int[] nums,int startIndex){
        result.add(new ArrayList<Integer>(path));
        for(int i=startIndex;i<nums.length;i++){
            path.add(nums[i]);
            bracktrack(nums,i+1);
            path.removeLast();
        }
    }

}
*题目9:90. 子集 II

思路:
1)递归法

class Solution {
    List<List<Integer>> result =new ArrayList<>();
    LinkedList<Integer> path=new LinkedList<>();
    public List<List<Integer>> subsetsWithDup(int[] nums) {
        if(nums.length==0){
           return result;
        }
        int[] used=new int[nums.length];
        Arrays.sort(nums);
        bracktrack(nums,0,used);
        return result;
    }
    void bracktrack(int[] nums,int startIndex,int[] used){
        result.add(new ArrayList<Integer>(path));
        for(int i=startIndex;i<nums.length;i++){
        	//树层去重 0就是false
            if(i>0&&nums[i]==nums[i-1]&&used[i-1]==0){
                continue;
            }
            path.add(nums[i]);
            used[i]=1;
            bracktrack(nums,i+1,used);
            path.removeLast();
             used[i]=0;
        }
    }
}
*题目10:491. 递增子序列

思路:
1)递归法

class Solution {
    List<List<Integer>> res = new ArrayList<>();
    List<Integer> path = new ArrayList<>();
    public List<List<Integer>> findSubsequences(int[] nums) {
        backTracking(nums, 0);
        return res;
    }
    private void backTracking(int[] nums, int startIndex){
        //题目要求至少有两个元素
        if(path.size() >= 2) res.add(new ArrayList<>(path));            
        HashSet<Integer> hs = new HashSet<>();
        for(int i = startIndex; i < nums.length; i++){
            //如果最后一个元素大于此元素 就不能取了 或者是取到的元素在同一层已经用过了
            if(!path.isEmpty() && path.get(path.size() -1 ) > nums[i] || hs.contains(nums[i]))
                continue;
            hs.add(nums[i]);
            path.add(nums[i]);
            backTracking(nums, i + 1);
            path.remove(path.size() - 1);
        }
    }
}

排列问题

题目11:46. 全排列

思路:与以上问题不同的是排列问题不用startIndex参数了,因为排列问题,每次都要从头开始搜索,例如元素1在[1,2]中已经使用过了,但是在[2,1]中还要再使用一次1,而used数组,其实就是记录此时path里都有哪些元素使用了,一个排列里(递归里)一个元素只能使用一次
1)递归法

class Solution {
    List<List<Integer>> result=new ArrayList<>();
    LinkedList<Integer> path=new LinkedList<>();
    public List<List<Integer>> permute(int[] nums) {
        if(nums.length==0) return result;
        boolean[] used=new boolean[nums.length];
        backTrack(nums,used);
        return result;
    }
    public void backTrack(int[] nums,boolean[] used){
        if(path.size()==nums.length){
            result.add(new ArrayList<Integer>(path));
            return;
        }
        for(int i=0;i<nums.length;i++){
            if(used[i]==true){
                continue;
            }
            path.add(nums[i]);
            used[i]=true;
            backTrack(nums,used);
            used[i]=false;
            path.removeLast();
        }
    }
}
*题目12:47. 全排列 II

思路:used[i-1]==false是树层上去重,如果需要树枝上去重used[i-1]==true 是同一树枝 这题与子集II与组合总和 II题目 去重逻辑相同
1)递归法

class Solution {
    List<List<Integer>> result=new ArrayList<>();
    LinkedList<Integer> path=new LinkedList<>();
    public List<List<Integer>> permuteUnique(int[] nums) {
        if(nums.length==0){
            return result;
        }
        boolean[] used=new boolean[nums.length];
        Arrays.sort(nums);
        backTrack(nums,used);
        return result;
    }
    public void backTrack(int[]nums,boolean[] used){
        if(nums.length==path.size()){
            result.add(new ArrayList<Integer>(path));
            return;
        }
        for(int i=0;i<nums.length;i++){
        	//去重逻辑
            if(i>0&&nums[i]==nums[i-1]&&used[i-1]==false){
                continue;
            }
            path.add(nums[i]);
            used[i]=true;
            backTrack(nums,used);
            path.removeLast();
            used[i]=false;
        }
    }
}
题目13:332. 重新安排行程

思路:
1)递归法

class Solution {
    private LinkedList<String> res = new LinkedList<>();
    private 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");
        //记录车票是否使用过
        boolean[] used = new boolean[tickets.size()];
        backTracking((ArrayList)tickets, used);
        return res;
    }
    public boolean backTracking(ArrayList<List<String>> tickets, boolean[] used) {
        // 机场的个数=航班数+1 证明找到了一条路径
        if (path.size() == tickets.size() + 1) {
            res = path;
            return true;
        }

        for (int i = 0; i < tickets.size(); i++) {
            //车票没用过并且车票地点与收集的路径到达地点相等 为了保证起始位置
            if (!used[i] && tickets.get(i).get(0).equals(path.getLast())) {
                //把到达地点添加到路径中
                path.add(tickets.get(i).get(1));
                //表示车票已经用过
                used[i] = true;
                if (backTracking(tickets, used)) {
                    return true;
                }
                used[i] = false;
                path.removeLast();
            }
        }
        return false;
    }
}

棋盘问题

题目14:51. N 皇后

思路:
1)递归法

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) {
        //如果递归的row 等于棋盘的行 就收集结果
        if (row == n) {
            List<String> list = new ArrayList<>();
            for (char[] c : chessboard) {
                list.add(String.copyValueOf(c));
            }
            res.add(list);
            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 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;
    }
}
题目15:37. 解数独

思路:
1)递归法

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;
                }
                // (i, j) 这个位置放k是否合适
                for (char k = '1'; k <= '9'; k++){
                    if (isValidSudoku(i, j, k, board)){
                        board[i][j] = k;
                        if (solveSudokuHelper(board)){ // 如果找到合适一组立刻返回
                            return true;
                        }
                        board[i][j] = '.';
                    }
                }
                // 9个数都试完了,都不行,那么就返回false
                return false;
            }
        }
        // 遍历完没有返回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;
    }
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值