zoukankan      html  css  js  c++  java
  • 34. 在排序数组中查找元素的第一个和最后一个位置

    给定一个按照升序排列的整数数组 nums,和一个目标值 target。找出给定目标值在数组中的开始位置和结束位置。

    你的算法时间复杂度必须是 O(log n) 级别。

    如果数组中不存在目标值,返回 [-1, -1]。

    示例 1:

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

    示例 2:

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

    来源:力扣(LeetCode)
    链接:https://leetcode-cn.com/problems/find-first-and-last-position-of-element-in-sorted-array


    思路:

    这种题目的解决方式肯定不唯一,这里归到了二分查找了那就还是用二分查找来解决吧

    可以用二分查找找出第一个位置和最后一个位置,但是寻找的方法有所不同,需要实现两个二分查找。

    我们将寻找 target 最后一个位置,转换成寻找 target+1 第一个位置,再往前移动一个位置。

    比如说找8,那我先找一个8第一个位置,再找一个9的第一个位置,第一个例子中没有9,所以会定位到10的位置前一个就是8的结尾位置

    这样我们只需要实现一个二分查找代码即可。

    class Solution {
        public int[] searchRange(int[] nums, int target) {
            int first = findFirst(nums, target);
            int last = findFirst(nums, target + 1) - 1;
            if (first == nums.length || nums[first] != target) {
                return new int[]{-1, -1};
            } else {
                return new int[]{first, Math.max(first, last)};
            }
        }
    
        private int findFirst(int[] nums, int target) {
            int l = 0, h = nums.length; // 注意 h 的初始值
            while (l < h) {
                int m = l + (h - l) / 2;
                if (nums[m] >= target) {
                    h = m;
                } else {
                    l = m + 1;
                }
            }
            return l;
        }
    }
  • 相关阅读:
    BZOJ1006: [HNOI2008]神奇的国度
    弦图与区间图
    后缀自动机
    插头DP
    BZOJ3328: PYXFIB
    BZOJ2118: 墨墨的等式
    BZOJ3916: [Baltic2014]friends
    BZOJ1337: 最小圆覆盖
    BZOJ3784: 树上的路径
    代码模版
  • 原文地址:https://www.cnblogs.com/zzxisgod/p/13339506.html
Copyright © 2011-2022 走看看