2055. 蜡烛之间的盘子
给你一个长桌子,桌子上盘子和蜡烛排成一列。给你一个下标从 0 开始的字符串 s ,它只包含字符 ‘’ 和 ‘|’ ,其中 '’ 表示一个 盘子 ,’|’ 表示一支 蜡烛 。
同时给你一个下标从 0 开始的二维整数数组 queries ,其中 queries[i] = [lefti, righti] 表示 子字符串 s[lefti…righti] (包含左右端点的字符)。对于每个查询,你需要找到 子字符串中 在 两支蜡烛之间 的盘子的 数目 。如果一个盘子在 子字符串中 左边和右边 都 至少有一支蜡烛,那么这个盘子满足在 两支蜡烛之间 。
比方说,s = “|||||" ,查询 [3, 8] ,表示的是子字符串 "||**|” 。子字符串中在两支蜡烛之间的盘子数目为 2 ,子字符串中右边两个盘子在它们左边和右边 都 至少有一支蜡烛。
请你返回一个整数数组 answer ,其中 answer[i] 是第 i 个查询的答案。
示例 1:
输入:s = "**|**|***|", queries = [[2,5],[5,9]]
输出:[2,3]
解释:
- queries[0] 有两个盘子在蜡烛之间。
- queries[1] 有三个盘子在蜡烛之间。
示例 2:
输入:s = "***|**|*****|**||**|*", queries = [[1,17],[4,5],[14,17],[5,11],[15,16]]
输出:[9,0,0,0,0]
解释:
- queries[0] 有 9 个盘子在蜡烛之间。
- 另一个查询没有盘子在蜡烛之间。
提示:
- 3 <= s.length <= 1 0 5 10^5 105
- s 只包含字符 ‘*’ 和 ‘|’ 。
- 1 <= queries.length <= 1 0 5 10^5 105
- queries[i].length == 2
- 0 <= lefti <= righti < s.length
解题思路
维护一个前缀和数组,保存每根蜡烛的左边有多少个盘子,并且用数组记录下所有蜡烛的下标,因为我们是按序遍历数组,所以下标数组的元素都是递增的,对于每个查询queries[i] = [lefti, righti],我们使用二分差值找出下标数组中lefti右边的第一根蜡烛下标以及righti左边的第一根蜡烛的下标,利用前缀和数组就可以求出两根蜡烛间盘子的个数了
代码
class Solution {
public:
vector<int> platesBetweenCandles(string s, vector<vector<int>>& queries) {
int pre(0);
map<int,int> m;
vector<int> idx;
for (int i = 0; i < s.size(); ++i) {
if (s[i]=='*')
pre++;
else {
idx.push_back(i);
m[i]=pre;
}
}
vector<int> res;
for (auto q:queries){
int l=bs(idx,q[0]),r=bs(idx,q[1]);
if (l==r||q[0]==q[1]) {
res.push_back(0);
continue;
}
int a=idx.size();
//末尾
if(r>a-1)
r=a-1;
else
r= idx[r]==q[1]?r:r-1;
res.push_back(m[idx[r]]-m[idx[l]]);
//res.push_back(m[bs(idx,q[1])]-m[bs(idx,q[0])]);
}
return res;
}
int bs(vector<int> &a,int tar){
int l=0,r=a.size()-1;
while (l<=r){
int mid=(r-l)/2+l;
if(a[mid]>tar){
r=mid-1;
}else if(a[mid]<tar){
l=mid+1;
}else return mid;
}
return l;
}
};