题目描述
思路
前缀和
根据题意,找到左边界往右搜索的第一支蜡烛和右边界往左搜索的第一支蜡烛的index,计算出中间的盘子数,可以用前缀和解决。
Java实现
class Solution {
// 找到每个查询最左侧的一只蜡烛和最右侧的一只蜡烛,则在他们中间的盘子都是所求的在两只蜡烛之间的盘子。
// 可以使用前缀和解决。
public int[] platesBetweenCandles(String s, int[][] queries) {
int n = s.length();
int[] preSum = new int[n];
// 计算前缀和
for (int i = 0, sum = 0; i < n; i++) {
if (s.charAt(i) == '*') {
sum ++;
}
preSum[i] = sum;
}
// 预处理找到字符串所有位置中左侧第一支蜡烛
int[] left = new int[n];
for (int i = 0, l = -1; i < n; i++) {
if (s.charAt(i) == '|') {
l = i;
}
left[i] = l;
}
// 预处理找到字符串所有位置中右侧第一支蜡烛
int[] right = new int[n];
for (int i = n-1, r = -1; i >= 0; i--) {
if (s.charAt(i) == '|') {
r = i;
}
right[i] = r;
}
// 设置结果集
int[] ans = new int[queries.length];
for (int i = 0; i < queries.length; i++) {
int[] query = queries[i];
// l表示右边界往左遍历的第一支蜡烛,r表示左边姐往右遍历的第一支蜡烛
int l = left[query[1]], r = right[query[0]];
// 前缀和数组计算的是各个位置的当前盘子总数,所以需要使用右边界往左遍历的第一支蜡烛的index对应的前缀和减左边界往右遍历的第一支蜡烛的index对应的前缀和
ans[i] = l == -1 || r == -1 || r >= l ? 0 : preSum[l] - preSum[r];
}
// 返回结果集
return ans;
}
}
Python实现
class Solution:
# 找到左边界往右搜索到的第一支蜡烛和右边界往左搜索到的第一支蜡烛之间的盘子数
# 前缀和解决
def platesBetweenCandles(self, s: str, queries: List[List[int]]) -> List[int]:
n = len(s)
preSum, sumcur = [0] * n, 0
left, l, right, r = [0] * n, -1, [0] * n, -1
for i, ch in enumerate(s):
if ch == "*":
sumcur += 1
else:
l = i
preSum[i] = sumcur
left[i] = l
for i in range(n-1, -1, -1):
if s[i] == "|":
r = i
right[i] = r
ans = [0] * len(queries)
for i, (r, l) in enumerate(queries):
r, l = right[r], left[l]
if r >= 0 and l >= 0 and r < l:
ans[i] = preSum[l] - preSum[r]
return ans