aboutsummaryrefslogtreecommitdiffstats
path: root/35/main.py
blob: b78265d8df04e0afc7bfc1d15e5ba5e7ec2e2a09 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
#!/usr/bin/env python
from typing import List


class Solution:
    def searchInsert(self, nums: List[int], target: int) -> int:
        begin = 0
        end = len(nums) - 1
        mid = 0
        while begin <= end:
            mid = (end + begin) // 2
            if target < nums[mid]:
                end = mid - 1
            elif target > nums[mid]:
                begin = mid + 1
            else:
                return mid
        return begin


if __name__ == "__main__":
    solution = Solution()
    print(solution.searchInsert([1, 3, 5, 6], 5))
    print(solution.searchInsert([1, 3, 5, 6], 2))
    print(solution.searchInsert([1, 3, 5, 6], 7))