思路
1、使用一个数组记录c的位置
2、双指针,比较pos_c_index和pos_c_index下一个位置的abs的结果
代码
class Solution:
def shortestToChar(self, s: str, c: str) -> List[int]:
# 先使用一个数组记录c的位置,双指针
pos_c = []
for i in range(len(s)):
if s[i] == c:
pos_c.append(i)
dis_res = []
pos_c_index = 0
for i in range(len(s)):
if pos_c_index + 1 < len(pos_c):
if abs(i - pos_c[pos_c_index]) <= abs(i - pos_c[pos_c_index + 1]):
dis_res.append(abs(i - pos_c[pos_c_index]))
else:
dis_res.append(abs(i - pos_c[pos_c_index + 1]))
pos_c_index += 1
else: # pos_c_index在最后一个位置
dis_res.append(abs(i - pos_c[pos_c_index]))
return dis_res
复杂度
时间复杂度:o(n)
空间复杂度:o(n)->o(k),k为c出现的次数