821. 字符的最短距离

weixin_45882985

已于 2024-10-09 23:34:06 修改

阅读量3

点赞数

文章标签： python

于 2021-02-02 18:28:02 首次发布

本文链接：https://blog.csdn.net/weixin_45882985/article/details/113570287

版权

LeetCode题目专栏收录该内容

3 篇文章 0 订阅

订阅专栏

821. 字符的最短距离

给定一个字符串 S 和一个字符 C。返回一个代表字符串 S 中每个字符到字符串 S 中的字符 C 的最短距离的数组。

示例
输入：S = “loveleetcode”, C = ‘e’
输出：[3, 2, 1, 0, 1, 0, 0, 1, 2, 2, 1, 0]

先将字符串转为列表lst_S进行存储，再遍历lst_S列表，如果当前元素恰好是目标字符C，则距离为0，列表lst_S对应位置的值为0，如果不是目标字符，则对当前字符的左右两部分分别进行查找，查找到目标元素的位置，然后取较小值。

class Solution:
    def shortestToChar(self, S: str, C: str) -> List[int]:
        lst_S = list(S)
        for idx, e in enumerate(lst_S):
            if e == C:
                lst_S[idx] = 0
            else:
                l_str, r_str = S[:idx], S[idx + 1:]
                l_find_idx, r_find_idx = l_str.rfind(C), r_str.find(C)
                if l_find_idx == -1:
                    lst_S[idx] = r_find_idx + 1
                elif r_find_idx == -1:
                    lst_S[idx] = idx - l_find_idx
                else:
                    lst_S[idx] = min(r_find_idx + 1,idx - l_find_idx)
        return lst_S