LeetCode每日一题——821. 字符的最短距离

题目

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