Question

1848. 到目标元素的最小距离

English Version

题目描述

给你一个整数数组 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 <= 1000
• 1 <= nums[i] <= 104
• 0 <= start < nums.length
• target 存在于 nums 中

解法

方法一：一次遍历

遍历数组，找到所有等于 $target$ 的下标，然后计算 $|i - start|$，取最小值即可。

时间复杂度 $O(n)$，其中 $n$ 为数组 $nums$ 的长度。空间复杂度 $O(1)$。

class Solution:
  def getMinDistance(self, nums: List[int], target: int, start: int) -> int:
    return min(abs(i - start) for i, x in enumerate(nums) if x == target)

class Solution {
  public int getMinDistance(int[] nums, int target, int start) {
    int n = nums.length;
    int ans = n;
    for (int i = 0; i < n; ++i) {
      if (nums[i] == target) {
        ans = Math.min(ans, Math.abs(i - start));
      }
    }
    return ans;
  }
}

class Solution {
public:
  int getMinDistance(vector<int>& nums, int target, int start) {
    int n = nums.size();
    int ans = n;
    for (int i = 0; i < n; ++i) {
      if (nums[i] == target) {
        ans = min(ans, abs(i - start));
      }
    }
    return ans;
  }
};

func getMinDistance(nums []int, target int, start int) int {
  ans := 1 << 30
  for i, x := range nums {
    if t := abs(i - start); x == target && t < ans {
      ans = t
    }
  }
  return ans
}

func abs(x int) int {
  if x < 0 {
    return -x
  }
  return x
}

function getMinDistance(nums: number[], target: number, start: number): number {
  let ans = Infinity;
  for (let i = 0; i < nums.length; ++i) {
    if (nums[i] === target) {
      ans = Math.min(ans, Math.abs(i - start));
    }
  }
  return ans;
}

impl Solution {
  pub fn get_min_distance(nums: Vec<i32>, target: i32, start: i32) -> i32 {
    nums.iter()
      .enumerate()
      .filter(|&(_, &x)| x == target)
      .map(|(i, _)| ((i as i32) - start).abs())
      .min()
      .unwrap_or_default()
  }
}