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

因为有序
所以用二分法，分别找到第一个k和最后一个k的下标。时间O(logN)
  1. class Solution {
  2. public:
  3.     int GetNumberOfK(vector<int> data ,int k) {
  4.         int num=0;
  5.         int size=data.size();
  6.         if(size>0){
  7.             int num1=getfk(data,0,size-1,k);
  8.             int num2=getsk(data,0,size-1,k);
  9.             if(num1!=-1 &&num2!=-1)
  10.                 num=num2-num1+1;
  11.         }
  12.         return num;
  13.     }
  14.     int getfk(vector<int>&data,int l,int r,int k){
  15.         if(l>r)return -1;
  16.         int m=(l+r)>>1;
  17.         int m_val=data[m];
  18.         if(m_val>k)
  19.             r=m-1;
  20.         else if(m_val<k)
  21.             l=m+1;
  22.         else{
  23.             if((m>0 && data[m-1]!=k)|| m==0)
  24.                 return m;
  25.             else
  26.                 r=m-1;
  27.         }
  28.         return getfk(data,l,r,k);   
  29.     }
  30.     int getsk(vector<int>&data,int l,int r,int k){
  31.         if(l>r)return -1;
  32.         int m=(l+r)>>1;
  33.         int m_val=data[m];
  34.         if(m_val>k)
  35.             r=m-1;
  36.         else if(m_val<k)
  37.             l=m+1;
  38.         else{
  39.             if((m<data.size()-1 && data[m+1]!=k)|| m==data.size()-1)
  40.                 return m;
  41.             else
  42.                 l=m+1;
  43.         }
  44.         return getsk(data,l,r,k);   
  45.     }
  46. };





posted @ 2016-03-16 20:31  copperface  阅读(153)  评论(0)    收藏  举报