IMWeb提升营Day7 | 训练题37:数字在排序数组中出现的次数

来源:互联网 发布:linux echo 脚本 编辑:程序博客网 时间:2024/05/19 02:24

题目描述

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

思路一:暴力

/* * 思路一:暴力破解,没想到这也能过,看来测试用例也没有太大嘛 */class Solution {public:    int GetNumberOfK(vector<int> data ,int k) {        int cnt = 0;        for(int i = 0; i < data.size(); i++){            if(data[i] == k) cnt++;        }        return cnt;    }};

思路二

/* * 思路二:既然是有序数组,当然要找效率更高的算法啦,必然是二分法 */class Solution {public:    int GetNumberOfK(vector<int> data ,int k) {        int length = data.size();        //因为二分法必须要数量在2以及以上,所以先处理特殊的情况        if(length == 0) return 0;        if(length == 1 && data[0] == k) return 1;        int cnt = 0;        int index = -1, indexback;        int left = 0, right = data.size()-1;        //先找到相等的那个数的索引,题目没明说,但是这里是按照升序数组来处理的        while(left < right){             int mid = (left + right) / 2;            if(data[mid] == k){                index = mid;                break;            }else if(data[mid] > k){                right = mid - 1;            }else{                left = mid + 1;            }        }        //找到索引之后,往前后两个方向找相同数字,累计数量,不相同即可停止寻找        if(index >= 0){            indexback = index - 1;            while(indexback >= 0){                if(data[indexback] == k) cnt++;                else break;                indexback--;            }            while(index < length){                if(data[index] == k) cnt++;                else break;                index++;            }        }        return cnt;    }};
阅读全文
0 0