821. 字符的最短距离

821. 字符的最短距离

给你一个字符串 s 和一个字符 c ，且 c 是 s 中出现过的字符。
返回一个整数数组 answer ，其中 answer.length == s.length 且 answer[i] 是 s 中从下标 i 到离它 最近 的字符 c 的 距离 。
两个下标 i 和 j 之间的 距离 为 abs(i - j) ，其中 abs 是绝对值函数。

示例 1：
输入：s = “loveleetcode”, c = “e” 输出：[3,2,1,0,1,0,0,1,2,2,1,0] 解释：字符 ‘e’ 出现在下标 3、5、6 和 11 处（下标从 0 开始计数）。 距下标 0 最近的 ‘e’ 出现在下标 3 ，所以距离为 abs(0 - 3) = 3 。 距下标 1 最近的 ‘e’ 出现在下标 3 ，所以距离为 abs(1 - 3) = 2 。 对于下标 4 ，出现在下标 3 和下标 5 处的 ‘e’ 都离它最近，但距离是一样的 abs(4 - 3) == abs(4 - 5) = 1 。 距下标 8 最近的 ‘e’ 出现在下标 6 ，所以距离为 abs(8 - 6) = 2 。
提示：
示例 2：
输入：s = “aaab”, c = “b” 输出：[3,2,1,0]

• 1 <= s.length <= 104
• s[i] 和 c 均为小写英文字母
• 题目数据保证 c 在 s 中至少出现一次

思路

按题意模拟即可，遍历一次数组，记录下标，再遍历一次数组，找出最接近的下标

解答

/**
 * @param {string} S
 * @param {character} C
 * @return {number[]}
 */
var shortestToChar = function (S, C) {
  const arr = [];
  for (let i = 0; i < S.length; i++) {
    const char = S[i];
    if (char === C) arr.push(i);
  }
  const res = [];
  for (let i = 0; i < S.length; i++) {
    let min = Infinity;
    for (let j = 0; j < arr.length; j++) {
      min = Math.min(min, Math.abs(arr[j] - i));
    }
    res.push(min);
  }
  return res;
};