day2-821-字符的最短距离

思路

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出现的次数