码迷,mamicode.com
首页 > 编程语言 > 详细

【剑指offer】数组在排序数组中出现的次数

时间:2019-09-24 10:27:09      阅读:67      评论:0      收藏:0      [点我收藏+]

标签:方法   不同   复杂度   组元   pad   offer   double   add   dex   

题目描述

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

分析:数组有序,采用二分查找无疑

两种方法,时间复杂度差不多,都是利用二分查找,不过统计k出现的次数有所不同而已

方法1:二分查找k,找到任意一个k的下标index,index向两边扩展即可

方法2:二分查找k+0.5和k-0.5的插入位置index1和index2,因为数组元素都是整数,index1-index2就是k出现的次数

时间复杂度:O(log N)

方法1:

int GetNumberOfK(vector<int> v ,int k)
{
    if(v.size()==0)
        return 0;
    
    //查找任意一个k的位置index
    int low=0;
    int high=v.size()-1;
    int index=-1;
    while(low<=high)
    {
        int mid=(low+high)/2;
        if(v[mid]==k)
        {
            index=mid;
            break;
        }
        else if(v[mid]>k)
        {
            high=mid-1;
        }
        else if(v[mid]<k)
        {
            low=mid+1;
        }
    }
    if(index==-1)
        return 0;
        
    //index向两边扩展
    int c=0;
    for(int i=index-1; i>=0; i--)
    {
        if(v[i]==k)
            c++;
        else
            break;
    }
    int n=v.size();
    for(int i=index+1; i<n; i++)
    {
        if(v[i]==k)
            c++;
        else
            break;
    }
    return c+1;
}

方法2:

//查找k的插入位置
int f(vector<int> &v,double k)
{
    int low=0;
    int high=v.size()-1;
    while(low<=high)
    {
        int mid=(low+high)/2;
        if(v[mid]>k)
        {
            high=mid-1;
        }else if(v[mid]<k)
        {
            low=mid+1;
        }
    }
    return low;
}
int GetNumberOfK(vector<int> v ,int k)
{
    if(v.size()==0)
        return 0;
    int x=f(v,k+0.5);
    int y=f(v,k-0.5);
    if(x==-1||y==-1)
        return 0;
    //k+0.5和k+0.5的插入位置相减就是k出现次数
    return x-y;
}

【剑指offer】数组在排序数组中出现的次数

标签:方法   不同   复杂度   组元   pad   offer   double   add   dex   

原文地址:https://www.cnblogs.com/yinbiao/p/11576729.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!