821. 字符的最短距离
个人觉得昨天的这个题很经典.大家可以此题为基础练习多种算法思想, 为以后学习算法打基础.参考其它大佬的解法, 整理了2个不错的思路, 方便大家参考.
class Solution:
def shortestToChar(self, s: str, c: str) -> List[int]:
# 中心扩展法
len_s = len(s) - 1
arr = []
for i in range(len(s)):
shortest = float('inf')
if s[i] == c:
arr.append(0)
continue
else:
left, right = i, i
while left >= 0:
if s[left] == c:
shortest = min(shortest, i - left)
break
else:
left -= 1
while right <= len_s:
if s[right] == c:
shortest = min(shortest, right - i)
break
else:
right += 1
arr.append(shortest)
return arr
class Solution:
def shortestToChar(self, s: str, c: str) -> List[int]:
# 滑动窗口法
arr = []
left = float('-inf')
right = s.find(c)
for i in range(len(s)):
if i == right:
# 更新窗口
left = right
right = s.find(c, right + 1)
if s[i] == c:
arr.append(0)
continue
else:
# 加了个判断, 如果right = -1,即最后一个窗口的右边界为字符串长度时, 此时的最小距离为当前字符与左边界的距离!
arr.append(min(i - left, right - i)) if right != -1 else arr.append(i - left)
return arr
# find()方法
s = "abcdefb"
print(s.find("b")) # 1
print(s.find("b", 2)) # 6 从指定位置后, 查询下一个字符串
print(s.find("k")) # -1
# index方法
s = "abcdefb"
print(s.index("b")) ## 1
print(s.index("b", 2)) # 6
print(s.index("k")) # ValueError: substring not found
可见, find 和 index 使用方法基本相同
相同点
1.都可以查找字符串中第一个出现的字符
2.都可以指定从某一个索引后面开始, 查找下一个出现的字符
不同点
1.find 找不到元素时,会返回-1
2.index 找不到元素时, 会返回 ValueError
s = ['a', 'b', 'c', 'd', 'e', 'f', 'b']
print(s.index("b"))
print(s.index("b", 2)) # 6
注意
1.list中查找元素,只能使用index, 无find方法.
2.查找不到元素时, 一样会出现 ValueError的异常