leetcode-字符的最短距离

163 阅读2分钟

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

题目描述

给你一个字符串 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 中至少出现一次

思路

思路不难想,就是找到字串c在当前下标的左边和右边的第一个c,然后看那个距离比较短。
编码上,这里有个小技巧,例如我们计算距离左边最短c的距离的时候,我们通过用一个临时变量left保存到当前位置,左侧第一个c的下标,如果左侧没有c,那么此时ans[i]的值应该是无穷大,不过由于c在s中肯定出现,所以ans[i]取值范围为0~len-1,我们就可以用初始值len来表示这个无穷大。距离右边最短c的距离跟左边是类似的,不过我们给ans[i]赋值的时候要考虑给最小值,即ans[i] = Integer.min(ans[i], right-i)。当然,求左边的时候也应该这样,不过由于此时ans[i]=len,肯定比i-left更大,所以可以直接赋值。

Java版本代码

class Solution {
    public int[] shortestToChar(String s, char c) {
        int len = s.length();
        int[] ans = new int[len];
        Arrays.fill(ans, len);
        char[] chars = s.toCharArray();
        for (int i = 0, left = -1; i < len; i++) {
            if (chars[i] == c) {
                left = i;
            }
            if (left != -1) {
                ans[i] = i - left;
            }
        }
        for (int i = len - 1, right = len; i >= 0; i--) {
            if (chars[i] == c) {
                right = i;
            }
            if (right != len) {
                ans[i] = Integer.min(ans[i], right-i);
            }
        }
        return ans;
    }
}