给你一个字符串 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的位置,那么我们就很容易想到使用双指针的做法,一个左指针去找该字符左边最近出现的c的位置,一个右指针去找该字符右边最近出现的c的位置,然后一比较,最短的距离就是我们要的答案。
代码(Python):
class Solution(object):
def shortestToChar(self, s, c):
answer = [] #存储最后的结果
for i,str in enumerate(s):
left = i #左指针,找该字符左边的最近的字符c
right = i #右指针,找该字符右边的最近的字符c
while left >= 0:
if s[left] == c:
break
else:
left -= 1
if left < 0: #若left<0,说明该字符的左边没有出现字符c
left = 100000 #把left设置为一个很大的数,距离该字符最近的字符c一定取右边right的答案
while right <len(s):
if s[right] == c:
break
else:
right += 1
if right >= len(s): #若left>len(s),说明该字符的右边没有出现字符c
right = 100000 #把right设置为一个很大的数,距离该字符最近的字符c一定取左边left的答案
answer.append(min(abs(left-i),abs(right-i)))
return answer