剑指offer53 在排序数组中查找数字

46 阅读1分钟

哈希方法

int GetNumberOfK(vector<int>& nums, int k) {
        // write code here
     int result=0;
     map<int,int>m;


     for(int i=0;i<nums.size();i++)
     {
       m[nums[i]]++;
     }
     for(auto ch:m)
     {
      if(ch.first==k)
       result=ch.second;
     }


return result;
    }
};

但是注意越界问题和数组为空的问题:

    if(nums.size()==0||k==0)return result;

时间复杂度: 先全部遍历一遍,加到哈希表里O(n)

二分法

//因为data中都是整数,所以可以稍微变一下,不是搜索k的两个位置,而是搜索k-0.5和k+0.5
//这两个数应该插入的位置,然后相减即可。
class Solution {
public:
    int GetNumberOfK(vector<int> data ,int k) {
        return biSearch(data, k+0.5) - biSearch(data, k-0.5) ;
    }
private:
    int biSearch(const vector<int> & data, double num){
        int s = 0, e = data.size()-1;      
        while(s <= e){
            int mid = (e - s)/2 + s;
            if(data[mid] < num)
                s = mid + 1;
            else if(data[mid] > num)
                e = mid - 1;
        }
        return s;
    }
};

image.png

时间复杂度O(longn)