python 刷leetcode题目（37）

162. 寻找峰值

峰值元素是指其值大于左右相邻值的元素。

给定一个输入数组 nums，其中 nums[i] ≠ nums[i+1]，找到峰值元素并返回其索引。

数组可能包含多个峰值，在这种情况下，返回任何一个峰值所在位置即可。

你可以假设 nums[-1] = nums[n] = -∞。

示例 1:

输入: nums = [1,2,3,1]
输出: 2
解释: 3 是峰值元素，你的函数应该返回其索引 2。

示例 2:

输入: nums = [1,2,1,3,5,6,4]
输出: 1 或 5 
解释: 你的函数可以返回索引 1，其峰值元素为 2；
     或者返回索引 5， 其峰值元素为 6。

说明:

你的解法应该是 O(logN) 时间复杂度的。

代码：

class Solution(object):
    def findPeakElement(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """
        length = len(nums)
        for i in range(length):
            if i == 0 or i == length -1:
                continue
            else:
                if nums[i-1] < nums[i] and nums[i] > nums[i+1]:
                    return  i
        if nums[0] <= nums[-1]:
            return length -1
        else:
            return 0
        

34. 搜索范围

给定一个按照升序排列的整数数组 nums，和一个目标值 target。找出给定目标值在数组中的开始位置和结束位置。

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

如果数组中不存在目标值，返回 [-1, -1]。

示例 1:

输入: nums = [5,7,7,8,8,10], target = 8
输出: [3,4]

示例 2:

输入: nums = [5,7,7,8,8,10], target = 6
输出: [-1,-1]

代码如下：

class Solution(object):
    def searchRange(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: List[int]
        """
        length = len(nums)
        if length == 0 :
            return [-1, -1]
        #### 二分查找方法
        if nums[0] <= target and nums[-1] >= target:
            left, mid , right = 0, 0, length-1
            while(left <=right):   #### 注意临界条件， 这里是允许为左边等于右边的情况 nums = [1],target =1;返回结果应该是[0,0]
                mid = left + (right - left) // 2
                if nums[mid] == target:
                    left = mid
                    right = mid
                    while left-1 >= 0 and nums[left] == nums[left-1]:
                        left -= 1
                    while right +1 <= length-1 and nums[right] == nums[right+1]:
                        right += 1

                    return [left, right]

                if nums[mid] < target:
                    left += 1
                else:
                    right -= 1
        return [-1,-1]