E1848.到目标元素的最小距离
implementation, https://leetcode.cn/problems/minimum-distance-to-the-target-element/
给你一个整数数组 nums (下标 从 0 开始 计数)以及两个整数 target 和 start ,请你找出一个下标 i ,满足 nums[i] == target 且 abs(i - start) 最小化 。注意:abs(x) 表示 x 的绝对值。
返回 abs(i - start) 。
题目数据保证 target 存在于 nums 中。
示例 1:
输入:nums = [1,2,3,4,5], target = 5, start = 3
输出:1
解释:nums[4] = 5 是唯一一个等于 target 的值,所以答案是 abs(4 - 3) = 1 。示例 2:
输入:nums = [1], target = 1, start = 0
输出:0
解释:nums[0] = 1 是唯一一个等于 target 的值,所以答案是 abs(0 - 0) = 0 。示例 3:
输入:nums = [1,1,1,1,1,1,1,1,1,1], target = 1, start = 0
输出:0
解释:nums 中的每个值都是 1 ,但 nums[0] 使 abs(i - start) 的结果得以最小化,所以答案是 abs(0 - 0) = 0 。提示:
1 <= nums.length <= 10001 <= nums[i] <= 10^40 <= start < nums.lengthtarget存在于nums中
这是一个经典的数组遍历问题。题目要求在数组中找到所有等于 target 的元素,并计算它们的下标 i 与给定 start 之间的绝对距离,最后返回这些距离中的最小值。
解题思路
遍历数组:我们可以遍历整个数组
nums。匹配目标:检查当前元素
nums[i]是否等于target。计算距离:如果相等,计算当前下标与
start的距离,即abs(i - start)。更新最小值:维护一个变量来记录遍历过程中遇到的最小距离。
由于数组长度最大只有 1000,直接进行
的线性扫描效率非常高。
代码实现
python
class Solution:
def getMinDistance(self, nums: List[int], target: int, start: int) -> int:
# 初始化最小距离为一个很大的值
min_dist = float('inf')
# 遍历数组及其下标
for i, num in enumerate(nums):
# 如果找到目标元素
if num == target:
# 计算当前下标到 start 的距离
dist = abs(i - start)
# 更新最小距离
if dist < min_dist:
min_dist = dist
return min_dist优化写法(Pythonic)
利用 Python 的生成器表达式和 min 函数,可以将代码简化为一行:
python
class Solution:
def getMinDistance(self, nums: List[int], target: int, start: int) -> int:
return min(abs(i - start) for i, num in enumerate(nums) if num == target)复杂度分析
- 时间复杂度:
,其中 是数组 nums的长度。我们需要完整遍历一次数组。 - 空间复杂度:
,只使用了常数级别的额外空间来存储距离值。