标准库中next_permutation函数:找当前序列中元素排列的下一个排列,按照字典顺序进行排列。比如说数组排列"123",那么它的下一个排列为"132",并且返回true。如果当前序列没有下一个排列,我们返回false,且把当前排列置为最小的排列,比如说:排列"321",因为该排列已经是最大的排列,所以它没有下一个排列。我们把该排列置为"123",并且返回false。 标准库实现两个重载版本的next_permutation。第一个函数接受两个参数,并且返回一个bool值,第二个函数接受三个参数,也返回一个bool值。bool next_permutation(_BidIt _First, _BidIt _Last);//第一个参数指数组的开始位置的迭代器,第二个参数指数组结束位置的后一个位置的迭代器。 bool next_permutation(_BidIt _First, _BidIt _Last, _Pr _Pred);//前两个参数和第一个函数中参数表达的意思相同,第三个参数指数组中的元素按照何种序列进行比较。next_permutation函数的实现原理:我们把数组分割成三个划分,在第一个划分中的所有元素都不需要改变,第二个划分只有一个元素,并且该元素需要和第三个划分中的某个元素进行交换。在第三个划分中,所有元素都需要逆序处理。如图所示:
重点是我们如何把序列分割成三个划分。首先找到第三个划分和第二个划分的分界线。因为第三个划分按照由右向左按非降序排列,所以它们的分界线是第一对不满足非降序排列的元素中间的位置。第二个划分只有一个元素。剩余的元素自然就成为第一个划分。 把序列分割以后:我们需要对分割以后的划分进行操作。首先,在第三个划分中,找到第一个大于第二个划分中的元素的位置。比如上图中,在{5,3,1,0}中找到第一个大于{2}的元素的位置,最后找到的是元素3的位置。把2和3进行交换。然后把交换以后的第三个划分逆序。
代码实现如下:直接对序列进行排序:bool nextPermutation(vector<int> &nums) { int size = nums.size(); for (int i = size - 1; i > 0; --i) { //找到第二个划分与第一个划分的分界线 if (nums[i] > nums[i - 1]) { int val = nums[i - 1]; int j = size - 1; //找到第一大于val值的位置 for (; j >= i; --j) { if (nums[j] > val) break; } //交换val值和第一个大于val的值 swap(nums[i - 1], nums[j]); int l = i; int r = size - 1; //逆序处理 while (l < r) { swap(nums[l], nums[r]); l++; r--; } return true; } } int start = 0; int last = size - 1; while (start < last) { swap(nums[start], nums[last]); start++; last--; } return false; }标准库中还有prev_permutation函数,找到当前序列的前一个序列,比如说序列"132",返回的是"123"。实现代码如下:
bool prePermutation(vector<int> &nums) { int size = nums.size(); for (int i = size - 1; i > 0; --i) { if (nums[i] < nums[i - 1]) { int val = nums[i - 1]; int j = size - 1; for (; j >= i; --j) { if (nums[j] < val) break; } swap(nums[i - 1], nums[j]); int l = i; int r = size - 1; while (l < r) { swap(nums[l], nums[r]); l++; r--; } return true; } } int start = 0; int last = size - 1; while (start < last) { swap(nums[start], nums[last]); start++; last--; } return false; }
时间: 2024-10-11 20:05:13