数组是排序的,所以重复出现的数字是相邻排列的。
用二分查找算法,找到第一次出现的位置,和 最后一次出现的位置。
判断第一次出现的位置条件为:当前数字的前一个是否与之相等,若是则继续查找,否则该位置就是第一次出现的位置。
判断最后一次出现的位置条件为:当前数字的后一个是否与之相等,若是则继续查找,否则该位置就是最后一次出现的位置。
出现的次数= last - first +1
时间复杂度:O(logn)
class Solution { public: int GetNumberOfK(vector<int> data ,int k) { int result=0; if(data.empty()) return 0; int start = 0; int end = data.size()-1; int first = getFirstK(data,k,start,end); int last = getLastK(data,k,start,end); if(first >-1 && last >-1) result = last - first+1; return result; } int getFirstK(vector<int> data, int k, int start, int end) { if(start > end) return -1; int midIndex = (start + end)/2; int midData = data[midIndex]; if(midData == k) { if(midIndex > 0 && data[midIndex-1] != k || midIndex == 0) return midIndex; else end = midIndex-1; } else if(midData > k) end = midIndex-1; else start = midIndex+1; return getFirstK(data,k,start,end); } int getLastK(vector<int> data, int k, int start, int end) { if(start > end) return -1; int midIndex = (start + end)/2; int midData = data[midIndex]; if(midData == k) { if(midIndex < data.size()-1 && data[midIndex+1] != k || midIndex == data.size()-1) return midIndex; else start = midIndex+1; } else if(midData > k) end = midIndex-1; else start = midIndex+1; return getLastK(data,k,start,end); } };
原文地址:http://blog.csdn.net/tommyzht/article/details/47220975