[剑指Offer] 53_在排序数组中查找数字

题目1：数字在排序数组中出现的次数

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

例：

输入数组{1, 2, 3, 3, 3, 3, 4, 5}和数字3。由于3在这个数组中出现了4次，输出4。

---

思路

1. 遍历数组统计。
   1. 时间复杂度：O(n)
   2. 空间复杂度：O(1)
2. 遍历统计没有利用到这是个排序数组的信息。因为是排序数组，所以需要统计的数字i个数等于i_right - i_left + 1。只要找到数字的边界即可。同时可以利用二分法查找，先找到一个i，那么边界必然在i左右两侧，然后采取二分法查找，只要左边还是i说明，左边界在左区间，否则在右区间。寻找右边界同理。
   1. 时间复杂度：O(logn)
   2. 空间复杂度：O(1)

---

代码

思路2：时间复杂度：O(logn)，空间复杂度：O(1)

def num_of_k(nums, k):
    """
    :param nums: array
    :param k: find K
    :return: times that K appears
    """
    # find middle k
    left, right = 0, len(nums)-1
    while left <= right:
        mid = (left + right) // 2
        if nums[mid] < k:
            left = mid + 1
        elif nums[mid] > k:
            right = mid - 1
        else:
            break
    else:
        raise Exception('NO SUCH NUMBER')
    start_l, start_r = left, mid
    end_l, end_r = mid, right
    # find start boundary
    while start_l <= start_r:
        start = (start_l + start_r) // 2
        if nums[start] != k:
            start_l = start + 1
        else:
            if start > 0 and nums[start - 1] == k:
                start_r = start - 1
            else:
                break
    # find end boundary
    while end_l <= end_r:
        end = (end_l + end_r) // 2
        if nums[end] != k:
            end_r = end - 1
        else:
            if end + 1 < len(nums) and nums[end + 1] == k:
                end_l = end + 1
            else:
                break

    return end - start + 1

---

题目2：0~n-1中缺失的数字。

一个长度为n-1的递增排序数组中的所有数字都是唯一的，并且每个数字都在范围0~n-1之内。
在范围0~n-1内的n个数字中有且只有一个数字不在该数组中，请找出这个数字。

---

思路

1. 只需要检查数字与其下表是否一致。如果一致，说明左侧全，如果不一致，说明左侧缺数。
   1. 时间复杂度：O(logn)
   2. 空间复杂度：O(1)

---

代码

思路1：时间复杂度：O(logn)，空间复杂度：O(1)

def missing_num(nums):
    """
    :param nums: array
    :return: missing number
    """
    left, right = 0, len(nums) - 1
    while left <= right:
        mid = (left + right) // 2
        if nums[mid] == mid:
            left = mid + 1
        else:
            if nums[mid - 1] == mid - 1:
                return mid
            right = mid - 1
    return left

---

题目3：数组中数值和下标相等的元素。

假设一个单调递增的数组里的每个元素都是整数并且是唯一的。请编程实现一个函数，找出
数组中任意一个数值等于其下表的元素。

例：

在数组{-3, -1, 1, 3, 5}中，数字3和它的下标相等。

---

思路

1. 因为是递增数组，在n=i的左侧必然有n < i，右侧必然有n > i。
   1. 时间复杂度：O(logn)
   2. 空间复杂度：O(1)

---

代码

思路1：时间复杂度：O(logn)，空间复杂度：O(1)

def integer_identical_to_index(nums):
    """
    :param nums: array
    :return: the number equals its index
    """
    left, right = 0, len(nums) - 1
    while left <= right:
        mid = (left + right) >> 1
        if nums[mid] < mid:
            left = mid + 1
        elif nums[mid] > mid:
            right = mid - 1
        else:
            break
    else:
        raise Exception('NO SUCH NUMBER')
    return mid

---

思考

对于排序的数组常可以考虑用二分法来加速。