46. 全排列
难度中等
给定一个不含重复数字的数组 nums ,返回其 所有可能的全排列 。你可以 按任意顺序 返回答案。
示例 1:
输入:nums = [1,2,3]
输出:[[1,2,3],[1,3,2],[2,1,3],[2,3,1],[3,1,2],[3,2,1]]
示例 2:
输入:nums = [0,1]
输出:[[0,1],[1,0]]
示例 3:
输入:nums = [1]
输出:[[1]]
提示:
• 1 <= nums.length <= 6
• -10 <= nums[i] <= 10
• nums 中的所有整数 互不相同
来自 <https://leetcode-cn.com/problems/permutations/>
回溯
class Solution {
List<List<Integer>> res=new ArrayList<List<Integer>>();
public List<List<Integer>> permute(int[] nums)
{
List<Integer> list=new ArrayList<>();
backtrack(nums,list);
return res;
}
public void backtrack(int[] nums,List<Integer> list)
{
if(list.size()==nums.length)
{
res.add(new ArrayList(list));
return;
}
for(int i=0;i<nums.length;i++)
{
if(list.contains(nums[i]))
{
continue;
}
list.add(nums[i]);
backtrack(nums,list);
list.remove(list.size()-1);
}
}
}
优化
由于要排除不合法的选择,即不选结果集中已有的数,要用contains方法判断结果集中是否已有该数
由于contains方法需要O(n)时间的复杂度,因此想办法简化这个
思考是否能将当前已经选择的个数now传入参数,然后每次遍历,从第now个开始遍历就无需判断结果集中是否已有该数了
但这种想法会遇到一个问题
比如,需要你输出1 2 3的全排列
当你将 1 3 加入结果集时,你会从第三个数开始遍历,即又会将3加入结果集,且不会将2加入结果集
因此为了解决这一问题,在每次进行下一层遍历之前,将元素组中的下标为now的数和当前遍历到的数交换位置
在要取消选择之前,将位置换回来,就好了
即比如已经将1和3加入结果集了,此时将数组中2和3调换位置,那么下一层的遍历,就能直接从第三个数开始遍历,而此时nums中的第三个数就是我们要的2
class Solution {
List<List<Integer>> res=new ArrayList<List<Integer>>();
public List<List<Integer>> permute(int[] nums)
{
backtrack(nums,0,new ArrayList<Integer>());
return res;
}
public void backtrack(int[] nums,int now,List<Integer> output)
{
if(now==nums.length)
{
res.add(new ArrayList<Integer>(output));
return;
}
for(int i=now;i<nums.length;i++)
{
output.add(nums[i]);
change(nums,i,now);
backtrack(nums,now+1,output);
change(nums,i,now);
output.remove(output.size()-1);
}
}
public void change(int[] nums,int now,int i)
{
int temp=nums[i];
nums[i]=nums[now];
nums[now]=temp;
}
}