码迷,mamicode.com
首页 > 编程语言 > 详细

面试题38_数字在排序数组中出现的次数

时间:2015-08-03 01:17:16      阅读:173      评论:0      收藏:0      [点我收藏+]

标签:二分查找   重复出现   

题目描述

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

解题思路

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

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

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

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


出现的次数= 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);
    }
};









面试题38_数字在排序数组中出现的次数

标签:二分查找   重复出现   

原文地址:http://blog.csdn.net/tommyzht/article/details/47220975

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!