34. 在排序数组中查找元素的第一个和最后一个位置

2023-01-20  本文已影响0人  球果假水晶蓝

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

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

你必须设计并实现时间复杂度为 O(log n) 的算法解决此问题。

impl Solution {
    pub fn search_range(nums: Vec<i32>, target: i32) -> Vec<i32> {
        if nums.len()  == 0{
            return vec![-1,-1];
        }else{
            let mut left:usize = 0;
            let mut right = nums.len() - 1;
            while left <= right{
                if right > nums.len() - 1{ // 解决rust索引超出usize范围的问题
                    break;
                }
                let middle = left + (right -left) / 2;
                if nums[middle] < target{
                    left = middle + 1;
                }else if nums[middle] > target{
                    right = middle - 1;
                }else{  // 关键步骤
                    if nums[right] > target{
                        right -= 1;
                    }
                    if nums[left] < target{
                        left += 1;
                    }if nums[left] == target && nums[right] == target{
                        return vec![left as i32, right as i32];
                    }
                }
            }
        }
    return vec![-1,-1];
    }
}
https://leetcode.cn/problems/find-first-and-last-position-of-element-in-sorted-array/
上一篇下一篇

猜你喜欢

热点阅读