目录
1. 问题描述
给你一个字符串 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:
扫描二维码关注公众号,回复:
13829805 查看本文章
输入:s = "aaab", c = "b" 输出:[3,2,1,0]
提示:
1 <= s.length <= 104
s[i]
和c
均为小写英文字母- 题目数据保证
c
在s
中至少出现一次
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/shortest-distance-to-a-character
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
2. 解题分析
关键是字符c可能出现不止一次。
字符c将整个字符串分隔成了若干个区间。
假设c总共出现n次,记c第k次出现的位置为,则:
- 从s开头到为止的各下标i到最近的c的距离就是
- 从到s末尾的各下标i到最近的c的距离就是
- 在中间各区间中的各下标i到最近的c的距离就是
3. 代码实现
from typing import List
class Solution:
def shortestToChar(self, s: str, c: str) -> List[int]:
c_pos = []
for k in range(len(s)):
if s[k]==c:
c_pos.append(k)
# print(c_pos)
ans = len(s)*[0]
for i in range(c_pos[0]):
ans[i] = c_pos[0] - i
for i in range(c_pos[-1],len(s)):
ans[i] = i - c_pos[-1]
for k in range(0,len(c_pos)-1):
for i in range(c_pos[k],c_pos[k+1]):
# print(c_pos[k],c_pos[k+1])
ans[i] = min(i-c_pos[k],c_pos[k+1]-i)
return ans
if __name__ == "__main__":
sln = Solution()
s = "loveleetcode"
c = "e"
print(sln.shortestToChar(s, c))
s = "aaab"
c = "b"
print(sln.shortestToChar(s, c))
执行用时:40 ms, 在所有 Python3 提交中击败了78.89%的用户
内存消耗:15 MB, 在所有 Python3 提交中击败了44.03%的用户