leetcode第46题

104 阅读3分钟

本文已参与「新人创作礼」活动,一起开启掘金创作之路。

leetcode第46题

题目描述(中等难度)

image-20221019235815186

描述的很简单,就是给定几个数,然后输出他们所有排列的可能。

解法一 插入

这是自己开始想到的一个方法,考虑的思路是,先考虑小问题怎么解决,然后再利用小问题去解决大问题。没错,就是递归的思路。比如说,

如果只有 1 个数字 [ 1 ],那么很简单,直接返回 [ [ 1 ] ] 就 OK 了。

如果加了 1 个数字 2, [ 1 2 ] 该怎么办呢?我们只需要在上边的情况里,在 1 的空隙,也就是左边右边插入 2 就够了。变成 [ [ 2 1 ], [ 1 2 ] ]。

如果再加 1 个数字 3,[ 1 2 3 ] 该怎么办呢?同样的,我们只需要在上边的所有情况里的空隙里插入数字 3 就行啦。例如 [ 2 1 ] 在左边,中间,右边插入 3 ,变成 3 2 1,2 3 1,2 1 3。同理,1 2 在左边,中间,右边插入 3,变成 3 1 2,1 3 2,1 2 3,所以最后的结果就是 [ [ 3 2 1],[ 2 3 1],[ 2 1 3 ], [ 3 1 2 ],[ 1 3 2 ],[ 1 2 3 ] ]。

如果再加数字也是同样的道理,只需要在之前的情况里,数字的空隙插入新的数字就够了。

思路有了,直接看代码吧。

 public List<List<Integer>> permute(int[] nums) {
     return permute_end(nums,nums.length-1);
 }
 // end 表示当前新增的数字的位置
 private List<List<Integer>> permute_end(int[] nums, int end) {
     // 只有一个数字的时候
     if(end == 0){
         List<List<Integer>> all = new ArrayList<>();
         List<Integer> temp = new ArrayList<>();
         temp.add(nums[0]);
         all.add(temp);
         return all;
     }
     //得到上次所有的结果
     List<List<Integer>> all_end = permute_end(nums,end-1);
     int current_size = all_end.size();
     //遍历每一种情况
     for (int j = 0; j < current_size; j++) { 
         //在数字的缝隙插入新的数字
         for (int k = 0; k <= end; k++) {
             List<Integer> temp = new ArrayList<>(all_end.get(j));
             temp.add(k, nums[end]);
             //添加到结果中
             all_end.add(temp);
         };
 ​
     }
     //由于 all_end 此时既保存了之前的结果,和添加完的结果,所以把之前的结果要删除
     for (int j = 0; j < current_size; j++) {
         all_end.remove(0);
     }
     return all_end;
 }
 Copy

既然有递归的过程,我们也可以直接改成迭代的,可以把递归开始不停压栈的过程省略了。

 public List<List<Integer>> permute(int[] nums) {
     List<List<Integer>> all = new ArrayList<>();
     all.add(new ArrayList<>());
     //在上边的基础上只加上最外层的 for 循环就够了,代表每次新添加的数字
     for (int i = 0; i < nums.length; i++) {
         int current_size = all.size();
         for (int j = 0; j < current_size; j++) {
             for (int k = 0; k <= i; k++) {
                 List<Integer> temp = new ArrayList<>(all.get(j));
                 temp.add(k, nums[i]);
                 all.add(temp);
             }
         }
         for (int j = 0; j < current_size; j++) {
             all.remove(0);
         }
     }
     return all;
 }
 Copy

时间复杂度,如果只分析代码的话挺复杂的。如果从最后的结果来说,应该是 n! 个结果,所以时间复杂度应该是 O(n!)。

空间复杂度:O(1)。

解法二 回溯

这个开始没想到,参考这里

其实也算是蛮典型的回溯,利用递归每次向 temp 里添加一个数字,数字添加够以后再回来进行回溯,再向后添加新的解。

可以理解成一层一层的添加,每一层都是一个 for 循环。

image-20221019235836482

每调用一层就进入一个 for 循环,相当于列出了所有解,然后挑选了我们需要的。其实本质上就是深度优先遍历 DFS。

 public List<List<Integer>> permute(int[] nums) {
    List<List<Integer>> list = new ArrayList<>(); 
    backtrack(list, new ArrayList<>(), nums);
    return list;
 }
 ​
 private void backtrack(List<List<Integer>> list, List<Integer> tempList, int [] nums){
    if(tempList.size() == nums.length){
       list.add(new ArrayList<>(tempList));
    } else{
       for(int i = 0; i < nums.length; i++){ 
          if(tempList.contains(nums[i])) continue; // 已经存在的元素,跳过
          tempList.add(nums[i]); //将当前元素加入
          backtrack(list, tempList, nums); //向后继续添加
          tempList.remove(tempList.size() - 1); //将 tempList 刚添加的元素,去掉,尝试新的元素
       }
    }
 }
 Copy

时间复杂度:

空间复杂度:

解法三 交换

参考这里

这个想法就很 cool 了,之前第一个解法的递归,有点儿动态规划的意思,把 1 个数字的解,2 个数字的解,3 个数字的解,一环套一环的求了出来。

假设有一个函数,可以实现题目的要求,即产生 nums 的所有的组合,并且加入到 all 数组中。不过它多了一个参数,begin,即只指定从 nums [ begin ] 开始的数字,前边的数字固定不变。

 upset(int[] nums, int begin, List<List<Integer>> all)
 Copy

如果有这样的函数,那么一切就都简单了。

如果 begin 等于 nums 的长度,那么就表示 begin 前的数字都不变,也就是全部数字不变,我们只需要把它加到 all 中就行了。

 if (begin == nums.length) {
     ArrayList<Integer> temp = new ArrayList<Integer>(); 
     for (int i = 0; i < nums.length; i++) {
         temp.add(nums[i]);
     }
     all.add(new ArrayList<Integer>(temp));
     return;
 }
 Copy

如果是其它的情况,我们其实只需要用一个 for 循环,把每一个数字都放到 begin 一次,然后再变化后边的数字就够了,也就是调用 upset 函数,从 begin + 1 开始的所有组合。

 for (int i = begin; i < nums.length; i++) {
     swap(nums, i, begin);
     upset(nums, begin + 1, all);
     swap(nums, i, begin);
 }
 Copy

总体就是这样了。

 public List<List<Integer>> permute(int[] nums) {
     List<List<Integer>> all = new ArrayList<>();
     //从下标 0 开始的所有组合
     upset(nums, 0, all);
     return all;
 }
 ​
 private void upset(int[] nums, int begin, List<List<Integer>> all) {
     if (begin == nums.length) {
         ArrayList<Integer> temp = new ArrayList<Integer>(); 
         for (int i = 0; i < nums.length; i++) {
             temp.add(nums[i]);
         }
         all.add(new ArrayList<Integer>(temp));
         return;
     }
     for (int i = begin; i < nums.length; i++) {
         swap(nums, i, begin);
         upset(nums, begin + 1, all);
         swap(nums, i, begin);
     }
 ​
 }
 ​
 private void swap(int[] nums, int i, int begin) {
     int temp = nums[i];
     nums[i] = nums[begin];
     nums[begin] = temp;
 }
 Copy

时间复杂度:

空间复杂度:

总结

这道题很经典了,用动态规划,回溯,递归各实现了一遍,当然解法一强行递归了一下,和解法三相比真是相形见绌,解法三才是原汁原味的递归,简洁优雅。