利用python 完成leetcode31 下一个排列
实现获取下一个排列的函数,算法需要将给定数字序列重新排列成字典序中下一个更大的排列。
如果不存在下一个更大的排列,则将数字重新排列成最小的排列(即升序排列)。
必须原地修改,只允许使用额外常数空间。
以下是一些例子,输入位于左侧列,其相应输出位于右侧列。 1,2,3 → 1,3,2 3,2,1 → 1,2,3 1,1,5 → 1,5,1 分析 当数组降序排列时,按升序排序即可 其他情况,将数组从后向前遍历,当左侧字符小于当前字符时结束 例如 1 3 5 9 7 6 遍历到i =3 nums[i]=9时结束,,9 与5 交换的排列必然比当前排列大,所以前面数字可以不考虑,而根据之前遍历数组 第i个之后的部分必然是按降序排列,找到后半部分最小,且比nums[i]大的元素,与nums[i]交换,再将后半部分设置为最小排列即可 代码
def nextPermutation(self, nums): i = len(nums) - 1 while i>0 and nums[i] <= nums[i - 1]: i -= 1 if(i == 0): nums.sort() return j = i - 1 while i < len(nums) - 1 and nums[i+1] > nums[j]: i += 1 nums[i],nums[j] = nums[j],nums[i] nums[j+1:] = sorted(nums[j+1:])
下一篇:
数据结构 - 树的种类及其应用