题目
给你一个字符串 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]
思路
维持一个滑动窗口,设置左右边界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