数字在排序数组中出现的次数

时间限制:C/C++ 1秒,其他语言2秒 空间限制:C/C++ 32M,其他语言64M

题目描述

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

思路:
1、暴力搜索,遍历数组中的元素,相同则计数加1

class Solution {
public:
    int GetNumberOfK(vector<int> data ,int k) {
        int count = 0;
        for(int i = 0;i < data.size();i++)
        {
            if(data[i] == k)
                count++;
        }
        return count;
    }
};

2、由于该数组为排序数组(但未指明是升序还是降序,这里姑且认为他是升序的),对其进行二分搜索,找到这个元素后,停止搜索,以该元素为基准,向前向后搜索,相等计数加1

class Solution {
public:
    int GetNumberOfK(vector<int> data ,int k) {
        int low = 0;
        int high = data.size()-1;
        int middle = 0;
        while(low < high)
        {
            middle = low + (high - low)/2;
            if(data[middle] == k)
            {
                break;
            }
            else if(data[middle] > k)
            {
                high = middle -1;
            }
            else
            {
                low = middle + 1;
            }
        }
        int count = 0;
        //向前向后寻找
        if(low <= high)
        {
            for(int i= middle;i >= 0;i--)
                if(data[i] == k)
                    count++;
                else
                    break;
            for(int i = middle+1;i<data.size();i++)
                if(data[i] == k)
                    count++;
                else
                    break;
        }
        return count;
    }
};

3、也可以采用二分搜索,找到该元素第一个出现的位置,以及最后出现的位置,最后相减得到出现的次数

public class Solution {
    public int GetNumberOfK(int [] array , int k) {
        if(array == null || array.length == 0)
            return 0;
        int first = getFirstK(array,k,0,array.length - 1);
        int last = getLastK(array,k,0,array.length - 1);
        if(first == -1 || last == -1)
            return 0;
        else
            return last - first + 1;
    }
    private int getFirstK(int [] array, int k, int low, int high){
        int mid = 0;
        while(low <= high){
            mid = low + (high -low) / 2;
            if(array[mid] == k){
                if(mid > 0 && array[mid - 1] != k || mid == 0)
                    return mid;
                else
                    high = mid - 1;
            }
            else if(array[mid] > k)
                high = mid - 1;
            else
                low = mid + 1;
        }
        return -1;
    }
    private int getLastK(int [] array, int k, int low, int high){
        int mid = 0;
        while(low <= high){
            mid = low + (high -low) / 2;
            if(array[mid] == k){
                if(mid < array.length - 1 && array[mid + 1] != k || mid == array.length - 1)
                    return mid;
                else
                    low = mid + 1;
            }
            else if(array[mid] > k)
                high = mid - 1;
            else
                low = mid + 1;
        }
        return -1;
    }
}

4、这个方法思路很奇特,虽同样采用二分,但特点在于既然目的找int 类型的k值,那找k-0.5,k+0.5两个浮点数的位置(大于k+0.5的位置和小于k-0.5的位置),然后相减,即为所得

class Solution {
public:
    int GetNumberOfK(vector<int> data ,int k) {
        return binSearch(data,k+0.5) - binSearch(data,k-0.5);
    }
    int binSearch(vector<int> &data,double num)
    {
        int low = 0;
        int high = data.size()-1;
        while(low <= high)
        {
            int mid = low + (high-low)/2;
            if(data[mid] < num)
                low = mid+1;
            else
                high = mid-1;
        }
        return high;
    }
};
posted @ 2020-03-31 10:28  牛犁heart  阅读(155)  评论(0)    收藏  举报