题目描述(中等难度)

46. Permutations - 图1

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

解法一 插入

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

如果只有 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 ] ]。

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

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

  1. public List<List<Integer>> permute(int[] nums) {
  2. return permute_end(nums,nums.length-1);
  3. }
  4. // end 表示当前新增的数字的位置
  5. private List<List<Integer>> permute_end(int[] nums, int end) {
  6. // 只有一个数字的时候
  7. if(end == 0){
  8. List<List<Integer>> all = new ArrayList<>();
  9. List<Integer> temp = new ArrayList<>();
  10. temp.add(nums[0]);
  11. all.add(temp);
  12. return all;
  13. }
  14. //得到上次所有的结果
  15. List<List<Integer>> all_end = permute_end(nums,end-1);
  16. int current_size = all_end.size();
  17. //遍历每一种情况
  18. for (int j = 0; j < current_size; j++) {
  19. //在数字的缝隙插入新的数字
  20. for (int k = 0; k <= end; k++) {
  21. List<Integer> temp = new ArrayList<>(all_end.get(j));
  22. temp.add(k, nums[end]);
  23. //添加到结果中
  24. all_end.add(temp);
  25. };
  26. }
  27. //由于 all_end 此时既保存了之前的结果,和添加完的结果,所以把之前的结果要删除
  28. for (int j = 0; j < current_size; j++) {
  29. all_end.remove(0);
  30. }
  31. return all_end;
  32. }

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

  1. public List<List<Integer>> permute(int[] nums) {
  2. List<List<Integer>> all = new ArrayList<>();
  3. all.add(new ArrayList<>());
  4. //在上边的基础上只加上最外层的 for 循环就够了,代表每次新添加的数字
  5. for (int i = 0; i < nums.length; i++) {
  6. int current_size = all.size();
  7. for (int j = 0; j < current_size; j++) {
  8. for (int k = 0; k <= i; k++) {
  9. List<Integer> temp = new ArrayList<>(all.get(j));
  10. temp.add(k, nums[i]);
  11. all.add(temp);
  12. }
  13. }
  14. for (int j = 0; j < current_size; j++) {
  15. all.remove(0);
  16. }
  17. }
  18. return all;
  19. }

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

空间复杂度:O(1)。

解法二 回溯

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

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

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

46. Permutations - 图2

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

  1. public List<List<Integer>> permute(int[] nums) {
  2. List<List<Integer>> list = new ArrayList<>();
  3. backtrack(list, new ArrayList<>(), nums);
  4. return list;
  5. }
  6. private void backtrack(List<List<Integer>> list, List<Integer> tempList, int [] nums){
  7. if(tempList.size() == nums.length){
  8. list.add(new ArrayList<>(tempList));
  9. } else{
  10. for(int i = 0; i < nums.length; i++){
  11. if(tempList.contains(nums[i])) continue; // 已经存在的元素,跳过
  12. tempList.add(nums[i]); //将当前元素加入
  13. backtrack(list, tempList, nums); //向后继续添加
  14. tempList.remove(tempList.size() - 1); //将 tempList 刚添加的元素,去掉,尝试新的元素
  15. }
  16. }
  17. }

时间复杂度:

空间复杂度:

解法三 交换

参考这里

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

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

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

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

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

  1. if (begin == nums.length) {
  2. ArrayList<Integer> temp = new ArrayList<Integer>();
  3. for (int i = 0; i < nums.length; i++) {
  4. temp.add(nums[i]);
  5. }
  6. all.add(new ArrayList<Integer>(temp));
  7. return;
  8. }

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

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

总体就是这样了。

  1. public List<List<Integer>> permute(int[] nums) {
  2. List<List<Integer>> all = new ArrayList<>();
  3. //从下标 0 开始的所有组合
  4. upset(nums, 0, all);
  5. return all;
  6. }
  7. private void upset(int[] nums, int begin, List<List<Integer>> all) {
  8. if (begin == nums.length) {
  9. ArrayList<Integer> temp = new ArrayList<Integer>();
  10. for (int i = 0; i < nums.length; i++) {
  11. temp.add(nums[i]);
  12. }
  13. all.add(new ArrayList<Integer>(temp));
  14. return;
  15. }
  16. for (int i = begin; i < nums.length; i++) {
  17. swap(nums, i, begin);
  18. upset(nums, begin + 1, all);
  19. swap(nums, i, begin);
  20. }
  21. }
  22. private void swap(int[] nums, int i, int begin) {
  23. int temp = nums[i];
  24. nums[i] = nums[begin];
  25. nums[begin] = temp;
  26. }

时间复杂度:

空间复杂度:

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

windliang wechat

添加好友一起进步~

如果觉得有帮助的话,可以点击 这里 给一个 star 哦 ^^

如果想系统的学习数据结构和算法,强烈推荐一个我之前学过的课程,可以点击 这里 查看详情