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

// 面试题53（一）：数字在排序数组中出现的次数
    // 题目：统计一个数字在排序数组中出现的次数。例如输入排序数组{1, 2, 3, 3,
    // 3, 3, 4, 5}和数字3，由于3在这个数组中出现了4次，因此输出4。
    static int get_count_of_num(const std::vector<int>& arr, int k)
    {
        int first = _get_start_of_num(arr, k, 0, arr.size() - 1);
        if(first > -1)
        {
            int last = _get_end_of_num(arr, k, first, arr.size() - 1);

            return last - first + 1;
        }

        return -1;
    }

    // 找到数组中第一个k的下标。如果数组中不存在k，返回-1
    static int _get_start_of_num(const std::vector<int>& arr, int k, uint32_t start, uint32_t end)
    {
        while (start <= end)
        {
            auto mid= (start + end) >> 1;

            if(arr[mid] == k)
            {
                if((mid > 0 && arr[mid - 1] != k) || mid == 0)
                {
                    return mid;
                }
                else
                {
                    end  = mid - 1;
                }
            }
            else if(arr[mid] > k)
            {
                end = mid - 1;
            }
            else
            {
                start = mid + 1;
            }
        }

        return -1;
    }

    // 找到数组中最后一个k的下标。如果数组中不存在k，返回-1
    static int _get_end_of_num(const std::vector<int>& arr, int k, uint32_t start, uint32_t end)
    {
        while (start <= end)
        {
            auto mid= (start + end) >> 1;

            if(arr[mid] == k)
            {
                if((mid < arr.size() - 1 && arr[mid + 1] != k) || mid == arr.size() - 1)
                {
                    return mid;
                }
                else
                {
                    start  = mid + 1;
                }
            }
            else if(arr[mid] > k)
            {
                end = mid - 1;
            }
            else
            {
                start = mid + 1;
            }
        }

        return -1;
    }

    // 面试题53（二）：0到n-1中缺失的数字
    // 题目：一个长度为n-1的递增排序数组中的所有数字都是唯一的，并且每个数字
    // 都在范围0到n-1之内。在范围0到n-1的n个数字中有且只有一个数字不在该数组
    // 中，请找出这个数字。
    static int get_missing_num(const std::vector<uint32_t>& arr)
    {
        uint32_t left = 0;
        uint32_t right = arr.size() - 1;
        while(left <= right)
        {
            uint32_t mid = (right + left) >> 1;
            if(arr[mid] != mid)
            {
                if(mid == 0 || arr[mid - 1] == mid - 1)
                {
                    return mid;
                }
                right = mid - 1;
            }
            else
            {
                left = mid + 1;
            }
        }

        return -1;
    }

    // 面试题53（三）：数组中数值和下标相等的元素
    // 题目：假设一个单调递增的数组里的每个元素都是整数并且是唯一的。请编程实
    // 现一个函数找出数组中任意一个数值等于其下标的元素。例如，在数组{-3, -1,
    // 1, 3, 5}中，数字3和它的下标相等。
    static bool get_num_eq_index(const std::vector<uint32_t>& arr, int& num)
    {
        uint32_t left = 0;
        uint32_t right = arr.size() - 1;
        while(left <= right)
        {
            uint32_t mid = ((right + left) >> 1);
            if(arr[mid] == mid)
            {
                num = mid;
                return true;
            }

            if(arr[mid] > mid)
            {
                right = mid - 1;
            }
            else
            {
                left = mid + 1;
            }
        }

        return false;
    }