目录
题目
给你一个字符串 s 和一个字符 c ,且 c 是 s 中出现过的字符。
返回一个整数数组 answer ,其中 answer.length == s.length 且 answer[i] 是 s 中从下标 i 到离它 最近 的字符 c 的 距离 。
两个下标 i 和 j 之间的 距离 为 abs(i - j) ,其中 abs 是绝对值函数。
示例
示例 1:
示例 2:
思路
维持一个滑动窗口,设置左右边界left、right,本题可分为三种情况
1、左边界不是指定字符右边界向右移动时遇到指定字符暂停,此时从left到right遍历下标,答案只能是右边界减去当前位置。最后答案添加完成后将左指针left跳到当前右指针上。
2、左边界有指定字符右边界向右移动时也遇到指定字符,此时从left到right遍历下标,答案是右边界减去当前位置和当前位置减去左边界的最小值,最后答案添加完成后将左指针left跳到当前右指针上。
3、左边界有指定字符,右边界跳到最终也未找到指定字符,此时从left到数组长度遍历下标,答案是当前位置减去左指针。
这种情况只会出现在最后一部分。
题解
def shortestToChar(self, s: str, c: str) -> List[int]:
# 初始化左右指针
left, right, res = 0, 1, []
while right < len(s):
# 未遇到指定字符,右指针向右移动
if s[right] != c:
right += 1
continue
# 第一种情况
elif s[left] == c:
for i in range(left, right):
res.append(min(abs(i-left), abs(i-right)))
left = right
right += 1
# 第二种情况
else:
for i in range(left, right):
res.append(abs(i-right))
left = right
right += 1
# 第三种情况
if len(res) < len(s):
for i in range(left, len(s)):
res.append(abs(i-left))
return res