#31 Next permutation
Permutations按照数值依次增大的次序排列,next permutation 就是找下一个大于当前数值的排列。
#解法一 std::next_permutation()
void nextPermutation(vector<int>& nums) {
next_permutation(nums.begin(), nums.end());
}
#解法二 std::next_permutation()的实现
- 从后往前开始遍历nums[], find the first i that nums[i-1] < nums[i]
- 如果没找到,则已经是最大的数了,reverse all,return最小的数
- 如果找到了,从后往前开始遍历,find nums[j+1] < nums[i] <nums[j]
- swap(nums[i], nums[j])
- reverse(nums[i+1], nums[n]) //n = nums.size()
class Solution {
public:
void nextPermutation(vector<int>& nums) {
int k = -1; //标志是否找到了i
for(int i = nums.size() - 2 ; i >= 0; --i){
if(nums[i] < nums[i+1]){
k = i;
break;
}
}
if(k == -1){
reverse(nums.begin(), nums.end());
return;
}
int l = -1;
for(int i = nums.size() - 1; i > k; i--){
if(nums[i] > nums[k]){
l = i;
break;
}
}
swap(nums[k], nums[l]);
reverse(nums.begin() + k + 1, nums.end());
}
};
#46 Permutations 不存在重复的数
#解法一 sort为升序序列 然后一直找next_permutation存入result中
class Solution {
public:
vector<vector<int>> permute(vector<int>& nums) {
//使用std::next_permutation()
vector<vector<int>> result;
sort(nums.begin(), nums.end()); //升序排列
do{
result.push_back(nums);
}while(next_permutation(nums.begin(), nums.end())); // 使用std::next_permutation()
return result;
}
};
#解法二 dfs回溯法(递归)
class Solution {
public:
vector<vector<int>> permute(vector<int>& nums) {
sort(nums.begin(), nums.end());
vector<vector<int>> result;
vector<int> path;
dfs(nums, path, result);
return result;
}
private:
void dfs(const vector<int> &nums, vector<int> path, vector<vector<int>> &result){
if(path.size() == nums.size()){
result.push_back(path);
return;
}
for(auto i : nums){
//pos 返回i在path中的位置索引
auto pos = find(path.begin(), path.end(), i);
if (pos == path.end()){ //i在path中没有出现
path.push_back(i);
dfs(nums, path, result);
path.pop_back();
}
}
}
};
#47 Permutations II 存在重复的数
跟46相比,用map记录出现的次数
class Solution {
public:
vector<vector<int>> permuteUnique(vector<int>& nums) {
vector<vector<int>> result;
vector<int> path;
map<int, int> map; //map<key, value>
for(int i : nums){
if(map.find(i) == map.end()) map[i] = 0; //map.find() 返回查找元素的位置,没有则返回map.end()
map[i]++;
}
permuteUnique(result, path, map, nums.size());
return result;
}
void permuteUnique(vector<vector<int>> &result, vector<int> &path, map<int, int> &map, int n)
{
if(n <= 0){
result.push_back(path);
return;
}
for(auto &p : map){
if(p.second <= 0) continue; //跳出当前循环,执行下一循环
p.second--;
path.push_back(p.first);
permuteUnique(result, path, map, n-1);
path.pop_back();
p.second++;
}
}
};
# Combinations