题目:
给你一个长桌子,桌子上盘子和蜡烛排成一列。给你一个下标从 0 开始的字符串 s ,它只包含字符 ‘’ 和 ‘|’ ,其中 '’ 表示一个 盘子 ,’|’ 表示一支 蜡烛 。
同时给你一个下标从 0 开始的二维整数数组 queries ,其中 queries[i] = [lefti, righti] 表示 子字符串 s[lefti…righti] (包含左右端点的字符)。对于每个查询,你需要找到 子字符串中 在 两支蜡烛之间 的盘子的 数目 。如果一个盘子在 子字符串中 左边和右边 都 至少有一支蜡烛,那么这个盘子满足在 两支蜡烛之间 。
- 比方说,s = “|||||" ,查询 [3, 8] ,表示的是子字符串 "||**|” 。子字符串中在两支蜡烛之间的盘子数目为 2 ,子字符串中右边两个盘子在它们左边和右边 都 至少有一支蜡烛。
请你返回一个整数数组 answer ,其中 answer[i] 是第 i 个查询的答案。
解答:
class Solution:
def platesBetweenCandles(self, s: str, queries: List[List[int]]) -> List[int]:
#暴力解法
m=len(s)
#count[i]:记录s[:i+1]中的*的个数
count=[0]*m
total=0
#left[i]:当前位置i左侧(含当前位置)的第一个蜡烛
left=[0]*m
lidx=-1
for i,c in enumerate(s):
if c=='*':
total+=1
else:
lidx=i
count[i]=total
left[i]=lidx
#right[i]:当前位置i右侧(含当前位置)的第一个蜡烛
right=[0]*m
ridx=-1
for i in range(m-1,-1,-1):
if s[i]=='|':
ridx=i
right[i]=ridx
n=len(queries)
res=[0]*n
for i,(start,end) in enumerate(queries):
#寻找区间内第一个蜡烛:start右侧的第一个蜡烛
first=right[start]
#寻找区间内最后一个蜡烛:end左侧第一个蜡烛
last=left[end]
if first==-1 or last==-1 or first>last:
continue
res[i]=count[last]-count[first]
return res