题目
给你一个长桌子,桌子上盘子和蜡烛排成一列。给你一个下标从 0 开始的字符串 s ,它只包含字符 ‘’ 和 ‘|’ ,其中 '’ 表示一个 盘子 ,’|’ 表示一支 蜡烛 。
同时给你一个下标从 0 开始的二维整数数组 queries ,其中 queries[i] = [lefti, righti] 表示 子字符串 s[lefti…righti] (包含左右端点的字符)。对于每个查询,你需要找到 子字符串中 在 两支蜡烛之间 的盘子的 数目 。如果一个盘子在 子字符串中 左边和右边 都 至少有一支蜡烛,那么这个盘子满足在 两支蜡烛之间 。
比方说,s = “|||||" ,查询 [3, 8] ,表示的是子字符串 "||**|” 。子字符串中在两支蜡烛之间的盘子数目为 2 ,子字符串中右边两个盘子在它们左边和右边 都 至少有一支蜡烛。
请你返回一个整数数组 answer ,其中 answer[i] 是第 i 个查询的答案。
示例
输入:s = “||***|”, queries = [[2,5],[5,9]]
输出:[2,3]
解释:
- queries[0] 有两个盘子在蜡烛之间。
- queries[1] 有三个盘子在蜡烛之间。
输入:s = “||||||*”, queries = [[1,17],[4,5],[14,17],[5,11],[15,16]]
输出:[9,0,0,0,0]
解释:
- queries[0] 有 9 个盘子在蜡烛之间。
- 另一个查询没有盘子在蜡烛之间。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/plates-between-candles
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
方法1:前缀和 + 二分
1、预处理数组
前缀和统计盘子数量
2、二分查找
将蜡烛的下标依次存入flag中
每次查询,在 前缀和数组flag 中使用二分查找,找到对应的左右边界(注意处理)。
Java实现
class Solution {
public int[] platesBetweenCandles(String s, int[][] queries) {
int n = queries.length;
int[] ans = new int[n];
int[] preSum = new int[s.length() + 1];
int kk = 0;
List<Integer> flag = new ArrayList<>();
char[] cs = s.toCharArray();
for (int i = 0; i < s.length(); i++) {
if (s.charAt(i) == '|') flag.add(i);
preSum[i + 1] = preSum[i] + (s.charAt(i) == '*' ? 1 : 0);
}
if (flag.size() == 0) return ans;
for (int[] q : queries) {
//左边界
int l = 0, r = flag.size() - 1;
int left = -1;
while (l < r) {
int mid = l + r >> 1;
if (flag.get(mid) >= q[0]) r = mid;
else l = mid + 1;
}
if (flag.get(l) >= q[0]) left = flag.get(l);
else {
ans[kk++] = 0;
continue;
}
//右边界
l = 0;
r = flag.size() - 1;
int right = -1;
while (l < r) {
int mid = l + r + 1 >> 1;
if (flag.get(mid) <= q[1]) l = mid;
else r = mid - 1;
}
if (flag.get(l) <= q[1]) right = flag.get(l);
else {
ans[kk++] = 0;
continue;
}
if (left <= right) ans[kk++] = preSum[right + 1] - preSum[left];
else {
ans[kk++] = 0;
}
}
return ans;
}
}