代码随想录：二分算法 leetcode704，leetcode35，leetcode34

# 二分算法:
# 重点：有序数组，无重复元素
#      注意二分算法边界问题
#      时间复杂度：O(logn)
#      空间复杂度：O(1)


# 题目：leetcode704
# 给定一个 n 个元素有序的（升序）整型数组 nums 和一个目标值 target  ，写一个函数搜索 nums 中的 target，如果目标值存在返回下标，否则返回 -1。
#
# 示例 1:
# 输入: nums = [-1,0,3,5,9,12], target = 9
# 输出: 4
# 解释: 9 出现在 nums 中并且下标为 4

class Solution:
    # 二分算法
    def binarySearch(self, nums, target):
        left = 0
        right = len(nums)-1
        while left <= right:
            mid = (left + right)//2
            if target < nums[mid]:
                right = mid - 1
            elif target > nums[mid]:
                left = mid + 1
            else:
                return mid
        return -1

    # leetcode35 :
    # 搜索插入位置：完整功能的二分法除了查找之外，还应该在没有查找到 target 元素时返回 target 应该插入的位置，为接下来可能的插入操作提供便利
    def searchInsert(self, nums, target):
        left = 0
        right = len(nums)-1
        while left <= right:
            mid = (left + right)//2
            if target < nums[mid]:
                right = mid - 1
            elif target > nums[mid]:
                left = mid + 1
            else:
                return mid
        return left


    # leetcode34 ：
    # 搜索插入位置：找第一次出现和最后一次出现的位置

    def binarySearchLowwer(self, nums, target):
        left = 0
        right = len(nums) - 1
        while left <= right:
            mid = (left + right) // 2
            if target <= nums[mid]:  # 如果target比mid值小，要在左边找，如果等于目标，也要在左边继续找还有没有一样的值
                right = mid - 1
            elif target > nums[mid]:
                left = mid + 1
        return left

    def binarySearchUpper(self, nums, target):
        left = 0
        right = len(nums) - 1
        while left <= right:
            mid = (left + right) // 2
            if target < nums[mid]:  # 如果target比目标值小，要在左边找
                right = mid - 1
            elif target >= nums[mid]: # 如果target比mid大，要在右边找；如果等于mid，也要在右边继续找还有没有一样的值
                left = mid + 1
        return right


if __name__ == '__main__':
    nums = [-1,-1,0,3,3,3,5,9,12]
    target = 3
    solution = Solution()
    print(solution.binarySearchUpper(nums, target))