一起养成写作习惯!这是我参与「掘金日新计划 · 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;
}
}