剑指Offer-在排序数组中查找数字 I

46.在排序数组中查找数字 I

统计一个数字在排序数组中出现的次数。

示例 1:

输入: nums = [5,7,7,8,8,10], target = 8
输出: 2
示例 2:

输入: nums = [5,7,7,8,8,10], target = 6
输出: 0

思路一

双指针。首尾各放一个指针,由于数组是排好序的,所以找到与target相等的那个范围即可。注意返回值,如果i和j相遇也没有找到target,那么就返回0,否则返回j-i+1。

class Solution {
    public int search(int[] nums, int target) {

        if(nums==null || nums.length==0 )
        {
            return 0;
        }
        int i=0;
        int j=nums.length-1;

        while(i<j)
        {
            if(nums[i]<target) i++;
            if(nums[j]>target) j--;
            if(nums[i]==target && nums[j]==target)
            {
                break;
            }
        }
        return nums[i]==target?(j-i+1):0;
    }
}

这种方法时间复杂度是O(N)。

思路二——二分

二分的精髓就是在已经排好序的数据中发现与寻找目标的联系。

数组已经排好序,如果找一个元素出现的次数,那么他们必然是相连的。当考虑相连的这段的端点时我们发现,左端点的左边都小于它,右边都大于等于它。右端点左边都小于等于它,右边都大于它。

利用这两个性质,我们可以使用两次二分,分别确定两个端点。端点找到了,用下标一减,就是次数。

class Solution {
    public int search(int[] nums, int target) {
        if(nums==null || nums.length==0)
        {
            return 0;
        }
        int l=0,r=nums.length-1;
        while(l<r)
        {
            int mid=(l+r)>>1;
            if(nums[mid]<target)
            {
                l=mid+1;
            }else
            {
                r=mid;
            }
        }
        if(nums[l]!=target) return 0;
        int left=l;
        l=0;
        r=nums.length-1;
        while(l<r)
        {
            int mid=(l+r+1)>>1;
            if(nums[mid]<=target)
            {
                l=mid;
            }else
            {
                r=mid-1;
            }
        }
        return r-left+1;
    }
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章