1. 程式人生 > >LeetCode-【陣列】-在排序陣列中查詢元素的第一個和最後一個位置

LeetCode-【陣列】-在排序陣列中查詢元素的第一個和最後一個位置

給定一個按照升序排列的整數陣列 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;
    }
}