描述
给定一个数组,例如[1,3,2],寻找下一个最小的大于当前数组组成的数字的数组,例如[2,1,3]
思路
双指针 + 从右向左找降序邻接对,“小数” + 从右向左找“大数” + 排序
想法:从右向左如果为升序,那么无法找到更大组合,所以需要找到降序元素,再对降序元素的较小数跟较大数进行替换,才能得到更大组合
1.从右向左寻找第一组邻接的降序对
2.如果没有找到,翻转数组,返回;如果找到,降序对的首元素作为“小数”,记录“小数”下标
3.从右向左寻找第一个比“小数”大的元素作为“大数”,记录“大数”下标
4.交换“小数”和“大数”元素
5.从小数+1到数组末尾,进行元素从小到大排序(sort)。
复杂度
时间:O(N)
空间:O(1)
代码(C++)
/* 【L31】下一个排列 */
bool cmp_small2large(int left, int right)
{
return left < right;
}
void swap_vec(std::vector<int>& nums, int left, int right)
{
int temp = nums[left];
nums[left] = nums[right];
nums[right] = temp;
}
void reverse(std::vector<int>& nums)
{
int start = 0;
int end = nums.size() - 1;
while (start < end)
{
swap_vec(nums, start, end);
start++;
end--;
}
return;
}
void nextPermutation(std::vector<int>& nums)
{
if (nums.size() <= 1) return;
// 从右向左寻找第一组邻接的降序对,减小的元素作为基数
int reverse_idx = nums.size() - 1;
bool found_reverse = false;
while (reverse_idx >= 1)
{
// 找到第一对邻接的降序对,跳出
if (nums[reverse_idx] > nums[reverse_idx - 1])
{
found_reverse = true;
break;
}
reverse_idx--;
}
if (found_reverse)
{
// 降序首元素作为“小数”
int small_idx = reverse_idx - 1;
int large_idx = reverse_idx;
// 从右向左寻找第一个比基数大的元素,作为交换基数位
for (int idx = nums.size() - 1; idx >= 0; idx--)
{
// 找到第一个“大数”,跳出
if (nums[idx] > nums[small_idx])
{
large_idx = idx;
break;
}
}
swap_vec(nums, small_idx, large_idx);
for (auto& ele : nums)
{
std::cout << ele;
}
std::cout << std::endl;
std::sort(nums.begin() + small_idx+1, nums.end(), cmp_small2large);
}
else
{
reverse(nums);
}
return;
}