剑指 Offer 53 - I. 在排序数组中查找数字 I

2021-07-16 LeetCode每日一题

链接:https://leetcode-cn.com/problems/zai-pai-xu-shu-zu-zhong-cha-zhao-shu-zi-lcof/

标签:数组、二分查找

题目

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

示例 1:

输入: nums = [5,7,7,8,8,10], target = 8
输出: 2

示例 2:

输入: nums = [5,7,7,8,8,10], target = 6
输出: 0

限制:

  • 0 <= 数组长度 <= 50000

分析

题目给定的是有序数组,所以除了直接循环遍历外,另一种方法就是二分查找,找到一个目标数的位置后,再往左右两步走,直到全部找到。

编码

class Solution {
    public int search(int[] nums, int target) {
        int len = nums.length;
        int left = 0, right = len - 1, count = 0, mid = 0;
        while (left <= right) {
            mid = left + (right - left) / 2;
            if (nums[mid] < target) {
                left = mid + 1;
            } else if (nums[mid] > target) {
                right = mid - 1;
            } else {
                break;
            }
        }

        int temp = mid;
        while (temp >=0 && temp < len && nums[temp] == target) {
            count++;
            temp--;
        }

        temp = mid + 1;
        while (temp >=0 && temp < len && nums[temp] == target) {
            count++;
            temp++;
        }

        return count;
    }
}
image-20210716202820092.png

你可能感兴趣的:(剑指 Offer 53 - I. 在排序数组中查找数字 I)