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

题目描述

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

解题思路

数组是排序的，所以重复出现的数字是相邻排列的。

用二分查找算法，找到第一次出现的位置，和 最后一次出现的位置。

判断第一次出现的位置条件为：当前数字的前一个是否与之相等，若是则继续查找，否则该位置就是第一次出现的位置。

判断最后一次出现的位置条件为：当前数字的后一个是否与之相等，若是则继续查找，否则该位置就是最后一次出现的位置。

出现的次数= last - first +1

时间复杂度：O(logn)

实现代码

[cpp]  view plain copy print ?

1. class Solution {  
2. public:  
3.     int GetNumberOfK(vector<int> data ,int k) {  
4.         int result=0;  
5.           
6.         if(data.empty())  
7.             return 0;  
8.         int start = 0;  
9.         int end = data.size()-1;  
10.           
11.         int first = getFirstK(data,k,start,end);  
12.         int last = getLastK(data,k,start,end);  
13.           
14.         if(first >-1 && last >-1)  
15.             result = last - first+1;  
16.           
17.         return result;  
18.     }  
19.       
20.     int getFirstK(vector<int> data, int k, int start, int end)  
21.     {  
22.         if(start > end)  
23.             return -1;  
24.         int midIndex = (start + end)/2;  
25.         int midData = data[midIndex];  
26.           
27.         if(midData == k)  
28.         {  
29.             if(midIndex > 0 && data[midIndex-1] != k || midIndex == 0)  
30.                 return midIndex;  
31.             else  
32.                 end = midIndex-1;  
33.         }  
34.         else if(midData > k)  
35.             end = midIndex-1;  
36.         else  
37.             start = midIndex+1;  
38.               
39.         return getFirstK(data,k,start,end);  
40.     }  
41.       
42.     int getLastK(vector<int> data, int k, int start, int end)  
43.     {  
44.         if(start > end)  
45.             return -1;  
46.         int midIndex = (start + end)/2;  
47.         int midData = data[midIndex];  
48.           
49.         if(midData == k)  
50.         {  
51.             if(midIndex < data.size()-1 && data[midIndex+1] != k || midIndex == data.size()-1)  
52.                 return midIndex;  
53.             else  
54.                 start = midIndex+1;  
55.         }  
56.         else if(midData > k)  
57.             end = midIndex-1;  
58.         else  
59.             start = midIndex+1;  
60.               
61.         return getLastK(data,k,start,end);  
62.     }  
63. };