Next Permutation

Question

Problem Statement

Given a list of integers, which denote a permutation.

Find the next permutation in ascending order.

Example

For [1,3,2,3], the next permutation is [1,3,3,2]

For [4,3,2,1], the next permutation is [1,2,3,4]

Note

The list may contains duplicate integers.

题解

找下一个升序排列，C++ STL 源码剖析一书中有提及，Permutations 一小节中也有详细介绍，下面简要介绍一下字典序算法：

  1. 从后往前寻找索引满足 a[k] < a[k + 1], 如果此条件不满足，则说明已遍历到最后一个。
  2. 从后往前遍历，找到第一个比a[k]大的数a[l], 即a[k] < a[l].
  3. 交换a[k]与a[l].
  4. 反转k + 1 ~ n之间的元素。

由于这道题中规定对于[4,3,2,1], 输出为[1,2,3,4], 故在第一步稍加处理即可。

Python

  1. class Solution(object):
  2.     def nextPermutation(self, nums):
  3.         """
  4.         :type nums: List[int]
  5.         :rtype: void Do not return anything, modify nums in-place instead.
  6.         """
  7.         if nums is None or len(nums) <= 1:
  8.             return
  9.         # step1: find nums[i] < nums[i + 1], Loop backwards
  10.         i = 0
  11.         for i in xrange(len(nums) - 2, -1, -1):
  12.             if nums[i] < nums[i + 1]:
  13.                 break
  14.             elif i == 0:
  15.                 # reverse nums if reach maximum
  16.                 nums = nums.reverse()
  17.                 return
  18.         # step2: find nums[i] < nums[j], Loop backwards
  19.         j = 0
  20.         for j in xrange(len(nums) - 1, i, -1):
  21.             if nums[i] < nums[j]:
  22.                 break
  23.         # step3: swap betwenn nums[i] and nums[j]
  24.         nums[i], nums[j] = nums[j], nums[i]
  25.         # step4: reverse between [i + 1, n - 1]
  26.         nums[i + 1:len(nums)] = nums[len(nums) - 1:i:-1]

C++

  1. class Solution {
  2. public:
  3.     /**
  4.      * @param nums: An array of integers
  5.      * @return: An array of integers that's next permuation
  6.      */
  7.     vector<int> nextPermutation(vector<int> &nums) {
  8.         if (nums.empty() || nums.size() <= 1) {
  9.             return nums;
  10.         }
  11.         // step1: find nums[i] < nums[i + 1]
  12.         int i = 0;
  13.         for (i = nums.size() - 2; i >= 0; --i) {
  14.             if (nums[i] < nums[i + 1]) {
  15.                 break;
  16.             } else if (0 == i) {
  17.                 // reverse nums if reach maximum
  18.                 reverse(nums, 0, nums.size() - 1);
  19.                 return nums;
  20.             }
  21.         }
  22.         // step2: find nums[i] < nums[j]
  23.         int j = 0;
  24.         for (j = nums.size() - 1; j > i; --j) {
  25.             if (nums[i] < nums[j]) break;
  26.         }
  27.         // step3: swap betwenn nums[i] and nums[j]
  28.         int temp = nums[i];
  29.         nums[i] = nums[j];
  30.         nums[j] = temp;
  31.         // step4: reverse between [i + 1, n - 1]
  32.         reverse(nums, i + 1, nums.size() - 1);
  34.         return nums;
  36.     }
  38. private:
  39.     void reverse(vector<int>& nums, int start, int end) {
  40.         for (int i = start, j = end; i < j; ++i, --j) {
  41.             int temp = nums[i];
  42.             nums[i] = nums[j];
  43.             nums[j] = temp;
  44.         }
  45.     }
  46. };

Java

  1. public class Solution {
  2.     /**
  3.      * @param nums: an array of integers
  4.      * @return: return nothing (void), do not return anything, modify nums in-place instead
  5.      */
  6.     public void nextPermutation(int[] nums) {
  7.         if (nums == null || nums.length == 0) return;
  9.         // step1: search the first nums[k] < nums[k+1] backward
  10.         int k = -1;
  11.         for (int i = nums.length - 2; i >= 0; i--) {
  12.             if (nums[i] < nums[i + 1]) {
  13.                 k = i;
  14.                 break;
  15.             }
  16.         }
  17.         // if current rank is the largest, reverse it to smallest, return
  18.         if (k == -1) {
  19.             reverse(nums, 0, nums.length - 1);
  20.             return;
  21.         }
  23.         // step2: search the first nums[k] < nums[l] backward
  24.         int l = nums.length - 1;
  25.         while (l > k && nums[l] <= nums[k]) l--;
  27.         // step3: swap nums[k] with nums[l]
  28.         int temp = nums[k];
  29.         nums[k] = nums[l];
  30.         nums[l] = temp;
  32.         // step4: reverse between k+1 and nums.length-1;
  33.         reverse(nums, k + 1, nums.length - 1);
  34.     }
  36.     private void reverse(int[] nums, int lb, int ub) {
  37.         for (int i = lb, j = ub; i < j; i++, j--) {
  38.             int temp = nums[i];
  39.             nums[i] = nums[j];
  40.             nums[j] = temp;
  41.         }
  42.     }
  43. }

源码分析

和 Permutation 一小节类似，这里只需要注意在step 1中i == -1时需要反转之以获得最小的序列。对于有重复元素，只要在 step1和 step2中判断元素大小时不取等号即可。Lintcode 上给的注释要求（其实是 Leetcode 上的要求）和实际给出的输出不一样。

复杂度分析

最坏情况下，遍历两次原数组，反转一次数组，时间复杂度为 O(n), 使用了 temp 临时变量，空间复杂度可认为是 O(1).