题目描述(中等难度)

又是一道全排列的题,之前在,46题,也讨论过全排列问题的一些解法。这道题的话,是给一个 n,不是输出它的全排列,而是把所有组合从从小到大排列后,输出第 k 个。

解法一

以 n = 4 为例,可以结合下图看一下。因为是从小到大排列,那么最高位一定是从 1 到 4。然后可以看成一组一组的,我们只需要求出组数,就知道最高位是多少了。而每组的个数就是 n - 1 的阶乘,也就是 3 的阶乘 6。

算组数的时候, 1 到 5 除以 6 是 0,6 除以 6 是 1,而 6 是属于第 0 组的,所有要把 k 减去 1。这样做除法结果就都是 0 了。

当然,还有一个问题下次 k 是多少了。求组数用的除法,余数就是下次的 k 了。因为 k 是从 1 计数的,所以如果 k 刚好等于了 perGroupNum 的倍数,此时得到的余数是 0 ,而其实由于我们求 groupNum 的时候减 1 了,所以此时 k 应该更新为 perGroupNum。

  1. k = k == 0 ? perGroupNum : k;

举个例子,如果 k = 6,那么 groupNum = ( k - 1 ) / 6 = 0, k % perGroupNum = 6 % 6 = 0,而下次的 k ,可以结合上图,很明显是 perGroupNum ,依旧是 6。

结合下图,确定了最高位属于第 0 组,下边就和上边的情况一样了。唯一不同的地方是最高位是 2 3 4,没有了 1。所有得到 groupNum 怎么得到最高位需要考虑下。

60. Permutation Sequence - 图1

综上,我们把它们整合在一起。

  1. public String getPermutation(int n, int k) {
  2. List<Integer> nums = new ArrayList<Integer>();
  3. for (int i = 0; i < n; i++) {
  4. nums.add(i + 1);
  5. }
  6. return getAns(nums, n, k);
  7. }
  8. private String getAns(List<Integer> nums, int n, int k) {
  9. if (n == 1) {
  10. //把剩下的最后一个数字返回就可以了
  11. return nums.get(0) + "";
  12. }
  13. int groupNum = (k - 1) / perGroupNum;
  14. nums.remove(groupNum);
  15. k = k % perGroupNum; //更新下次的 k
  16. k = k == 0 ? perGroupNum : k;
  17. return num + getAns(nums, n - 1, k);
  18. }
  19. public int factorial(int number) {
  20. if (number <= 1)
  21. return 1;
  22. else
  23. return number * factorial(number - 1);
  24. }

时间复杂度:

空间复杂度:

这是最开始自己的想法,有 3 点可以改进一下。

第 1 点,更新 k 的时候,有一句

很不优雅了,问题的根源就在于问题给定的 k 是从 1 编码的。我们只要把 k - 1 % perGroupNum,这样得到的结果就是 k 从 0 编码的了。然后求 groupNum = (k - 1) / perGroupNum; 这里 k 也不用减 1 了。

第 3 点,我们求 perGroupNum 的时候,每次都调用了求迭代的函数,其实没有必要的,我们只需要一次循环求出 n 的阶乘。然后在每次迭代中除以 nums 的剩余个数就够了。

综上,看一下优化过的代码吧。

  1. public String getPermutation(int n, int k) {
  2. List<Integer> nums = new ArrayList<Integer>();
  3. int factorial = 1;
  4. for (int i = 0; i < n; i++) {
  5. factorial *= i;
  6. }
  7. }
  8. factorial *= n; //先求出 n 的阶乘
  9. StringBuilder ans = new StringBuilder();
  10. k = k - 1; // k 变为 k - 1
  11. for (int i = n; i > 0; i--) {
  12. factorial /= (nums.size()); //更新为 n - 1 的阶乘
  13. int groupNum = k / factorial;
  14. int num = nums.get(groupNum);
  15. nums.remove(groupNum);
  16. k = k % factorial;
  17. ans.append(num);
  18. }
  19. return ans.toString();

时间复杂度:O(n),当然如果 remove 函数的时间是复杂度是 O(n),那么整体上就是 O(n²)。

空间复杂度:O(1)。

这道题其实如果写出来,也不算难,优化的思路可以了解一下。

添加好友一起进步~

如果想系统的学习数据结构和算法,强烈推荐一个我之前学过的课程,可以点击 查看详情