一起养成写作习惯！这是我参与「掘金日新计划 · 4 月更文挑战」的第17天，点击查看活动详情。

题目描述

给你一个字符串 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]

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/shortest-distance-to-a-character
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路分析

• 今天的算法题目是字符距离比较题目，题目容易理解，简述就是找到每个字符 c 的位置，然后比较每个字符 i 到 c 的距离，取最短的距离即位答案。
• 我们可以首先遍历字符串 s, 找到每一个 c 的位置，存储下来。然后遍历字符串 s 中每个字符和c的差值，实现代码如下：

通过代码

class Solution {
    public int[] shortestToChar(String s, char c) {
        int n = s.length();
        int[] ans = new int[n];
        List<Integer> cIdxArr = new ArrayList<>();
        for (int i = 0; i < n; i++) {
            if (s.charAt(i) == c) {
                cIdxArr.add(i);
            }
        }

        for (int i = 0; i < n; i++) {
            int temp = Integer.MAX_VALUE;
            for (int j = 0; j < cIdxArr.size(); j++) {
                temp = Math.min(temp, Math.abs(i - cIdxArr.get(j)));
                ans[i] = temp;
            }
        }

        return ans;
    }
}

总结

• 上述算法的时间复杂度是O(n * n), 空间复杂度是O(n)
• 坚持算法每日一题，加油！