下一个排列
实现获取下一个排列的函数,算法需要将给定数字序列重新排列成字典序中下一个更大的排列。
如果不存在下一个更大的排列,则将数字重新排列成最小的排列(即升序排列)。
必须原地修改,只允许使用额外常数空间。
以下是一些例子,输入位于左侧列,其相应输出位于右侧列。
1,2,3 → 1,3,2
3,2,1 → 1,2,3
1,1,5 → 1,5,1
我的思路:
也算是官方题解,基于下面这个图,一共有两次遍历,先从后往前找第一个升序,即下图a[i-1]>a[i](如果没找到,说明已经找到头了,整个数组都是降序排列,是最大的排列了,直接翻转整个数组,就是第一个最小的排列),这是第一次遍历。第二次遍历也是从后往前找,第一个比a[i-1]大的数,一定能找到这个数,因为至少a[i]比a[i-1]大,交换a[i-1]和a[j],再从a[i]到整个数组结尾翻转一下,就得到了结果。
代码一:
class Solution {
public:
void nextPermutation(vector<int>& nums) {
int i=nums.size()-2;
while(i>=0&&nums[i]>=nums[i+1])--i;//第一次遍历
if(i>=0){
int j=nums.size()-1;
while(1){
if(nums[j]>nums[i])break;//第二次遍历,不会死循环,我前面思路中有提到
--j;
}
swap(nums[j],nums[i]);
}
reverse(nums.begin()+i+1,nums.end());
}
};
代码二(algorithm 头文件自带排列):
class Solution {
public:
void nextPermutation(vector<int>& nums) {next_permutation(nums.begin(),nums.end());}
};
全排列
给定一个没有重复数字的序列,返回其所有可能的全排列。
示例:
输入: [1,2,3]
输出:
[
[1,2,3],
[1,3,2],
[2,1,3],
[2,3,1],
[3,1,2],
[3,2,1]
]
我的思路:
有了上面“下一个排列”的基础,求全排列只需要不断下一个排列即可,我们只需要计算需要多少次排列即可(即n!次排列)。
代码1:
class Solution {
public:
vector<vector<int>> permute(vector<int>& nums) {
vector<vector<int>> v;
int n=nums.size();
int k=1;
for(int i=1;i<=n;++i)
{
k*=i;
}
for(int i=0;i<k;++i)
{
v.push_back(nums);
next_permutation(nums.begin(),nums.end());
}
return v;
}
};
我的思路:
这是基于交换的全排列。
代码2:
class Solution {
public:
void dfs(vector<vector<int>> &res,vector<int> &nums,int k){
if(k==nums.size())res.push_back(nums);
for(int i=k;i<nums.size();++i)
{
swap(nums[i],nums[k]);
dfs(res,nums,k+1);
swap(nums[i],nums[k]);
}
}
vector<vector<int>> permute(vector<int>& nums) {
vector<vector<int>> res;
dfs(res,nums,0);
return res;
}
};
全排列 II
给定一个可包含重复数字的序列,返回所有不重复的全排列。
示例:
输入: [1,1,2]
输出:
[
[1,1,2],
[1,2,1],
[2,1,1]
]
我的思路:
基于上面代码2加上小小的判断去重即可。
代码:
class Solution {
public:
bool isswap(int i,int k,vector<int> &nums)
{
for(;i<k;++i)
if(nums[i]==nums[k])return false;
return true;
}
void dfs(vector<vector<int>> &res,vector<int> &nums,int k){
if(k==nums.size())res.push_back(nums);
for(int i=k;i<nums.size();++i)
{
if(!isswap(k,i,nums))continue;
swap(nums[i],nums[k]);
dfs(res,nums,k+1);
swap(nums[i],nums[k]);
}
}
vector<vector<int>> permuteUnique(vector<int>& nums) {
vector<vector<int>> res;
dfs(res,nums,0);
return res;
}
};