【剑指offer】面试题53:在排序数组中查找数字

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

  • 解题思路:
    遇到排序数组,优先无脑考虑二分法。找出这个数字的左边位置,再找出这个数字的右边位置,两个位置的距离就是出现的次数。复杂度是O(logn)。
  • 注意点
    注意边界判断,即没有找到的时候返回的是-1,此时应特殊输出0。
    【考点】 查找 二分法
class Solution {
public:
	int GetNumberOfK(vector<int> data, int k) {
		int length = data.size();
		int firstIndex = getFirstK(data, k, 0, length - 1);
		int lastIndex = getLastK(data, k, 0, length - 1);
		if (firstIndex == -1 || lastIndex == -1) return 0;
		int res = lastIndex - firstIndex + 1;
		return res;
	}
	int getFirstK(vector<int> data, int k, int low, int high) {
		if (low > high)
			return -1;

		int mid = (low + high) / 2;
		if (data[mid] == k)
		{
			if ((mid > 0 && data[mid - 1] != k) || mid == 0)
				return mid;
			else
				high = mid - 1;
		}
		else if (data[mid] > k)
			high = mid - 1;
		else
			low = mid + 1;
		return getFirstK(data, k, low, high);
	}
	int getLastK(vector<int> data, int k, int low, int high) {
		if (low > high)
			return -1;

		int mid = (low + high) / 2;
		if (data[mid] == k)
		{
			if ((mid < data.size()-1 && data[mid + 1] != k) || mid == data.size()-1)
				return mid;
			else
				low = mid + 1;
		}
		else if (data[mid] > k)
			high = mid - 1;
		else
			low = mid + 1;
		return getLastK(data, k, low, high);
	}
};

你可能感兴趣的:(算法,剑指offer,查找)