从零开始刷Leetcode——数组(31.33)

31. 下一个排列

实现获取下一个排列的函数，算法需要将给定数字序列重新排列成字典序中下一个更大的排列。
如果不存在下一个更大的排列，则将数字重新排列成最小的排列（即升序排列）。
必须原地修改，只允许使用额外常数空间。

例：123变成132；115变成151

1.排序+遍历

class Solution:
    def nextPermutation(self, nums: List[int]) -> None:
        """
        Do not return anything, modify nums in-place instead.
        """
        n = len(nums)
        flag = 0
        for i in range(len(nums)-1, 0, -1):
            if nums[i] > nums[i-1]:
                flag = 1
                nums[i:] = sorted(nums[i:])
                for j in range(i, len(nums)):
                    if nums[j]> nums[i-1]:
                        nums[i-1], nums[j] = nums[j], nums[i-1]
                        break
                break
        if flag == 0:
            nums.sort()

这道题逻辑是从后遍历，遇到前一个数比后一个数小的时候说明是非递减数组。这时只需要修改前一个数之后的数组即可，修改的方法是将该数之后的数字递增排序，将该数与排序之后第一个比该数大的数交换即可。40ms，13.7MB

33. 搜索旋转排序数组

假设按照升序排序的数组在预先未知的某个点上进行了旋转。

( 例如，数组 [0,1,2,4,5,6,7] 可能变为 [4,5,6,7,0,1,2] )。

搜索一个给定的目标值，如果数组中存在这个目标值，则返回它的索引，否则返回 -1 。

你可以假设数组中不存在重复的元素。

你的算法时间复杂度必须是 O(log n) 级别。

1.二分查找

class Solution:
    def search(self, nums: List[int], target: int) -> int:
        low = 0
        high = len(nums)-1
        if not nums: return -1
        while low < high-1:
            mid = (low+high)//2
            # 左边无旋转
            if nums[low] < nums[mid]:
                if target <= nums[mid] and target >= nums[low]:
                    high = mid
                else:
                    low = mid
            # 右边无旋转
            else:
                if target <= nums[high] and target >= nums[mid]:
                    low = mid
                else:
                    high = mid
        if nums[low] == target: return low
        if nums[high] == target: return high
        return -1

因为要求O(log n)，所以只能用二分查找，虽然数组经过旋转，但分割后前后两个数组必有一个是递增的，只需比较目标数与这个递增数组的首尾值就可以进行后续运算了。40ms，13.9MB