题目:
实现获取下一个排列的函数,算法需要将给定数字序列重新排列成字典序中下一个更大的排列。
如果不存在下一个更大的排列,则将数字重新排列成最小的排列(即升序排列)。
必须原地修改,只允许使用额外常数空间。
以下是一些例子,输入位于左侧列,其相应输出位于右侧列。
示例:
1,2,3
→1,3,2
3,2,1
→1,2,3
1,1,5
→1,5,1
此题要求原地修改,常熟空间,所以需要直接在元素组上进行元素的更改,从后向前遍历,找到第一个nums[i] < nums[i+1] 的 i 位置并记录,说明需要从此处进行变换。不过此时需要判断是否存在下一个更大的排列。若有,再次从后向前遍历找到第一个大于nums[i]的数,交换两个数的位置。此时nums[i]后面的数为非递增排列,只需反转就ok了。代码如下
C 代码:
void swap(int* a,int* b)
{
int temp = *a;
*a = *b;
*b = temp;
}
void reverse(int* nums,int i,int j) //反转i j范围内数组的元素
{
while (i < j)
{
swap(&nums[i],&nums[j]);
i++;
j--;
}
}
void nextPermutation(int* nums, int numsSize)
{
int k = -1,right = 0;
int i;
for (i = numsSize-2;i >= 0;i--)
{
if (nums[i+1] > nums[i])
{
k = i; //找到第一个小于后面的数 的元素
break;
}
}
if (k == -1) //不存在下一个更大的排列,直接反转
{
reverse(nums,0,numsSize-1);
return ;
}
for (i = numsSize-1;i >= 0;i--)
{
if (nums[i] > nums[k])
{
right = i;
break;
}
}
swap(&nums[k],&nums[right]);
reverse(nums,k+1,numsSize-1);
}
谢谢。