原题地址:https://leetcode-cn.com/problems/next-permutation/description/
题目描述:
实现获取下一个排列的函数,算法需要将给定数字序列重新排列成字典序中下一个更大的排列。
如果不存在下一个更大的排列,则将数字重新排列成最小的排列(即升序排列)。
必须原地修改,只允许使用额外常数空间。
以下是一些例子,输入位于左侧列,其相应输出位于右侧列。1,2,3
→ 1,3,2
3,2,1
→ 1,2,3
1,1,5
→ 1,5,1
题意看了半天才弄明白,数学中的排列组合,比如“1,2,3”的全排列,依次是:
1 2 3
1 3 2
2 1 3
2 3 1
3 1 2
3 2 1
- 1
- 2
- 3
- 4
- 5
- 6
是这样的排序方式。。
不难可以发现排序的规律,寻找最后一支子序列,其中第二个元素为子序列的最大值。不存在,则返回升序序列。存在,对子序列进行操作:以该子序列为目标,从第二个元素开始遍历,寻找比第一个数大的最小数,并与第一个数进行交换,最后再对剩下的序列进行升序排序。
解题方法:
class Solution { public: void nextPermutation(vector<int>& nums) { int n = nums.size(); int i, j = 0, k; int min, loc = 0; if(n == 0 || n == 1) return ; for(i = 1; i < n; i ++){ int flag = 1; min = nums[i]; for(k = i - 1; k < n; k ++){ if(nums[i] < nums[k]){ flag = 1; break; } if(nums[k] > nums[i - 1] && nums[k] <= min){ min = nums[k]; loc = k; flag = 0; } } if(flag == 0) j = i; } if(j != 0){ min = nums[loc]; nums[loc] = nums[j-1]; nums[j-1] = min; } sort(nums.begin() + j, nums.end()); } };
利用sort()函数进行排序:
sort(v.begin(),v.end());
默认排序方法是从小到大,若是从大到小或者自定义比较方式,需要自己改写comp()函数:
bool comp(const int &a,const int &b) { return a>b; } sort(v.begin(),v.end(),comp);