leetcode 31: 下一个排列

题目:

实现获取下一个排列的函数,算法需要将给定数字序列重新排列成字典序中下一个更大的排列。

如果不存在下一个更大的排列,则将数字重新排列成最小的排列(即升序排列)。

必须原地修改,只允许使用额外常数空间。

以下是一些例子,输入位于左侧列,其相应输出位于右侧列。
1,2,3 → 1,3,2
3,2,1 → 1,2,3
1,1,5 → 1,5,1

解析:

首先反向查找第一个比前一个数大的数字, 其下标为i, 然后从这个数字开始往后查找最小的比nums[i-1] 大的数字, 把它们交换以后在对nums[i]及之后的数字排序

 1 void nextPermutation(vector<int>& nums) {
 2     int i = nums.size()-1;
 3     for(; i >0; --i)
 4     {
 5         if(nums[i] > nums[i-1])
 6         {
 7             int m  = i;
 8             for(int j = i+1; j< nums.size();++j)
 9             {
10                 if(nums[j] > nums[i-1] && nums[j] <= nums[m])
11                     m =j;
12             }
13             swap(nums[m], nums[i-1]);
14             sort(nums.begin() + i, nums.end());
15             break;
16         }
17     }
18     if(i == 0)
19     {
20         sort(nums.begin(), nums.end());
21     }
22 }

猜你喜欢

转载自www.cnblogs.com/bzaq/p/9983644.html