LeetCode-【陣列】-在排序陣列中查詢元素的第一個和最後一個位置
阿新 • • 發佈:2019-02-08
給定一個按照升序排列的整數陣列 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]
題解:看到時間複雜度是O(log n) 級別,且陣列又是升序排列的,很容易想到二分查詢的方法,二分查詢在解決陣列有關的問題時有時真的是神出鬼沒的。
class Solution { public int[] searchRange(int[] nums, int target) { int n=nums.length; int[] res=new int[2]; res[0]=-1;res[1]=-1; int l=0,r=n-1,c=0; while(l<=r){ int mid=(l+r)/2; if(nums[mid]==target){ int i=mid; int j=mid; while(i>=0&&nums[i]==target) i--; res[0]=i<0?0:i+1; while(j<n&&nums[j]==target) j++; res[1]=j>n?n-1:j-1; return res; } if(nums[mid]<target) l=mid+1; else if(nums[mid]>target) r=mid-1; } return res; } }