leetcode 【 Search Insert Position 】python 实现
題目:
Given a sorted array 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 may assume no duplicates in the array.
Here are few examples.
[1,3,5,6], 5 → 2
[1,3,5,6], 2 → 1
[1,3,5,6], 7 → 4
[1,3,5,6], 0 → 0
?
代碼:oj測試通過?Runtime:?52 ms
1 class Solution: 2 # @param A, a list of integers 3 # @param target, an integer to be inserted 4 # @return integer 5 def search(self, A, start, end, target): 6 # search stop case I 7 if start == end: 8 if start == 0: 9 return [ 0,1 ][ A[0]<target ] 10 if A[start] == target: 11 return start 12 else: 13 return [start,start+1][A[start]<target] 14 # search stop case II 15 if start+1 == end: 16 if A[start] >= target: 17 return start 18 elif A[start] < target and A[end] >= target : 19 return end 20 else: 21 return end+1 22 23 mid = (start+end)/2 24 # search stop case III 25 if A[mid] == target: 26 return mid 27 if A[mid] > target: 28 return self.search(A, start, mid-1, target) 29 if A[mid] < target: 30 return self.search(A, mid+1, end, target) 31 32 def searchInsert(self, A, target): 33 # zero length case 34 if len(A) == 0: 35 return 0 36 # binary search 37 start = 0 38 end = len(A)-1 39 return self.search(A, start, end, target)思路:
二分查找經典題。
采用迭代方式時:
1. 考慮start==end的情況(一個元素)和start+1==end的情況(兩個元素),作為迭代終止的兩種case。
2. 當元素數量大于3時作為一般的case處理,二分查找。
3. 根據題意要求進行判斷條件。
4. 第一次提交沒有AC ,原因是在處理start==end的case時候,竟然只考慮了0和len(A)的邊界情況,沒有考慮一般情況,陷入了思維的陷阱。
?
后面又寫了一版非遞歸的代碼:oj測試通過?Runtime:?63 ms
1 class Solution: 2 # @param A, a list of integers 3 # @param target, an integer to be inserted 4 # @return integer 5 def searchInsert(self, A, target): 6 # zero length case 7 if len(A) == 0 : 8 return 0 9 # binary search 10 start = 0 11 end = len(A)-1 12 while start <= end : 13 if start == end: 14 if start == 0: 15 return [0,1][A[0]<target] 16 if A[start] == target: 17 return start 18 else: 19 return [start,start+1][A[start]<target] 20 if start+1 == end: 21 if A[start] >= target: 22 return start 23 elif A[start] < target and A[end] >= target: 24 return end 25 else: 26 return end+1 27 mid = (start+end)/2 28 if A[mid] == target: 29 return mid 30 elif A[mid] > target: 31 end = mid - 1 32 else: 33 start = mid + 1思路跟非遞歸差不太多。
個人感覺判斷stop case的代碼雖然邏輯上比較清晰(剩一個元素或者兩個元素或者直接找到了target),但是并不是很簡潔。后續再不斷改進。
轉載于:https://www.cnblogs.com/xbf9xbf/p/4245711.html
總結
以上是生活随笔為你收集整理的leetcode 【 Search Insert Position 】python 实现的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: java的循环中try catch的一点
- 下一篇: 一篇很好的架构文章