回溯法(+DFS):空间换时间(LC46)

回溯法的核心是回溯。在搜索到某一节点的时候,如果我们发现目前的节点(及其子节点)并不是需求目标时,我们回退到原来的节点继续搜索,并且把在目前节点修改的状态还原。这样的好处是我们可以始终只对图的总状态进行修改,而非每次遍历时新建一个图来储存状态。

在具体的写法上,它与普通的深度优先搜索一样,都有 [修改当前节点状态]→[递归子节
点] 的步骤,只是多了回溯的步骤,变成了 [修改当前节点状态]→[递归子节点]→[回改当前节点
状态]。

结合题目加代码注释去理解回溯更加清晰

class Solution {
    public List<List<Integer>> permute(int[] nums) {
         int len=nums.length;
         //使用一个动态数组res保存所有可能的全排列
         List<List<Integer>> res=new ArrayList<>();
          
          if(len==0){
              return res;
          }

          boolean[] used=new boolean[len];
          List<Integer> path=new ArrayList<>();

          dfs(nums,len,0,path,used,res);
          return res;
    }
     //path:往下走一层的时候,path 变量在尾部追加,
      //而往回走的时候,需要撤销上一次的选择,也是在尾部操作,因此 path 变量是一个栈;

      //depath(index):表示当前程序递归到第几层,表示当前要确定的是某个全排列中下标为 index 的那个数是多少;
    
    //布尔数组 used,初始化的时候都为 false 表示这些数还没有被选择,当我们选定一个数的时候,就将这个数组的相应位置设置为 true
    //考虑下一个位置的时候,就能够以 O(1)O(1) 的时间复杂度判断这个数是否被选择过,这是一种「以空间换时间」的思想。
    private void dfs(int[]nums,int len,int depath,List<Integer>path,boolean[]used,List<List<Integer>> res ){

    if(depath==len){
        res.add(new ArrayList<>(path));//递归结束条件,就是当depath和数组一样,说明找到一组排列
        return;
    }

    // 在非叶子结点处,产生不同的分支,这一操作的语义是:在还未选择的数中依次选择一个元素作为下一个位置的元素,这显然得通过一个循环实现。
    for(int i=0;i<len;i++){
        if(!used[i]){//未被选择
        path.add(nums[i]);//往栈path里添加未选择的元素
        used[i]=true;//并且将标记为选择过
        

       //找一个等价表达式,就是要向下一层移动depath+1
       dfs(nums,len,depath+1,path,used,res);
        // 注意:下面这两行代码发生 「回溯」,回溯发生在从 深层结点 回到 浅层结点 的过程,代码在形式上和递归之前是对称的
         //上面递归结束,就回溯到上一次
         used[i]=false;
         path.remove(path.size()-1);

    }
}
    }
}
全部评论

相关推荐

点赞 收藏 评论
分享
牛客网
牛客企业服务