-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy path35.py
55 lines (45 loc) · 1.35 KB
/
35.py
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
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
from typing import List
# Runtime: 48 ms, faster than 69.84% of Python3 online submissions for Search Insert Position.
# Memory Usage: 14.9 MB, less than 92.97% of Python3 online submissions for Search Insert Position.
class Solution:
def searchInsert(_, nums: List[int], target: int) -> int:
try:
return nums.index(target)
except:
nums += [target]
nums.sort()
return nums.index(target)
"""
35. Search Insert Position
Given a sorted array of distinct integers and a target value,
return the index if the target is found.
If not, return the index where it would be if it were inserted in order.
You must write an algorithm with O(log n) runtime complexity.
Example 1:
Input: nums = [1,3,5,6], target = 5
Output: 2
Example 2:
Input: nums = [1,3,5,6], target = 2
Output: 1
Example 3:
Input: nums = [1,3,5,6], target = 7
Output: 4
Example 4:
Input: nums = [1,3,5,6], target = 0
Output: 0
Example 5:
Input: nums = [1], target = 0
Output: 0
Constraints:
1 <= nums.length <= 104
-104 <= nums[i] <= 104
nums contains distinct values sorted in ascending order.
-104 <= target <= 104
"""
# Testcase
s = Solution()
print(s.searchInsert([1,3,5,6], 5)) # 2
print(s.searchInsert([1,3,5,6], 2)) # 1
print(s.searchInsert([1,3,5,6], 7)) # 4
print(s.searchInsert([1,3,5,6], 0)) # 0
print(s.searchInsert([1], 0)) # 0