821. Shortest Distance to a Character

Posted 2021-01-15 bernieloveslife

Given a string S and a character C, return an array of integers representing the shortest distance from the character C in the string.

Example 1:

Input: S = "loveleetcode", C = ‘e‘
Output: [3, 2, 1, 0, 1, 0, 0, 1, 2, 2, 1, 0]

Note:

1. S string length is in [1, 10000].
2. C is a single character, and guaranteed to be in string S.
3. All letters in S and C are lowercase.

class Solution:
    def shortestToChar(self, S, C):
        """
        :type S: str
        :type C: str
        :rtype: List[int]
        """
        ans = []
        for i in range(len(S)):
            if S[i]==C:
                ans.append(i)
        res = []
        for i in range(len(S)):
            temp = [abs(j-i) for j in ans]
            res.append(min(temp))
        return res