- 蜡烛之间的盘子
给你一个长桌子,桌子上盘子和蜡烛排成一列。给你一个下标从 0 开始的字符串 s ,它只包含字符 ‘*’ 和 ‘|’ ,其中 '’ 表示一个 盘子 ,’|’ 表示一支 蜡烛 。
同时给你一个下标从 0 开始的二维整数数组 queries ,其中 queries[i] = [lefti, righti] 表示 子字符串 s[lefti…righti] (包含左右端点的字符)。对于每个查询,你需要找到 子字符串中 在 两支蜡烛之间 的盘子的 数目 。如果一个盘子在 子字符串中 左边和右边 都 至少有一支蜡烛,那么这个盘子满足在 两支蜡烛之间 。
比方说,s = “||**||**|*” ,查询 [3, 8] ,表示的是子字符串 "||**|" 。子字符串中在两支蜡烛之间的盘子数目为 2 ,子字符串中右边两个盘子在它们左边和右边 都 至少有一支蜡烛。
请你返回一个整数数组 answer ,其中 answer[i] 是第 i 个查询的答案。
1.前缀和
前缀和数组presum[i]记录i位置之前(包括这个位置)有多少个盘子。
nearLeft[i]记录i位置左边最近的蜡烛的位置;nearRight[i]记录i位置右边最近的蜡烛的位置。左边没有蜡烛可以记为-1,右边没有蜡烛可以记为n
queries[i][0]表示查询的起始位置,记为start,queries[i][1]记录查询的结束位置,记为end。则start右边最近的蜡烛是第一根蜡烛,end左边最近的蜡烛是最后一根蜡烛。
class Solution {
public:
vector<int> platesBetweenCandles(string s, vector<vector<int>>& queries) {
int n=s.size();
vector<int>nearLeft(n),nearRight(n),presum(n);
vector<int>res(queries.size());
int left=-1,right=n;
presum[0]=s[0]=='*';
for(int i=1;i<n;i++){
if(s[i]=='*'){
presum[i]=presum[i-1]+1;
}else{
presum[i]=presum[i-1];
}
}
for(int i=0;i<n;i++){
if(s[i]=='|'){
left=i;
}
nearLeft[i]=left;
if(s[n-1-i]=='|'){
right=n-1-i;
}
nearRight[n-1-i]=right;
}
for(int i=0;i<queries.size();i++){
int start=nearRight[queries[i][0]],end=nearLeft[queries[i][1]];
if(start<n&&end>=0){
res[i]=max(presum[end]-presum[start],0);
}
}
return res;
}
};
2.二分法
将蜡烛的下标保存到数组中,查找时使用二分查找
class Solution {
public:
vector<int> platesBetweenCandles(string s, vector<vector<int>>& queries) {
vector<int>candles;//记录每根蜡烛的位置
vector<int>res(queries.size());//结果
vector<int>presum(s.size());//前缀和数组
presum[0]=s[0]=='*';
for(int i=1;i<s.size();i++){
presum[i]=presum[i-1]+(s[i]=='*');
}
for(int i=0;i<s.size();i++){
if(s[i]=='|'){
candles.push_back(i);
}
}
for(int i=0;i<queries.size();i++){
int start=queries[i][0],end=queries[i][1];
int left=-1,right=-1;
int l=0,r=candles.size()-1,mid;
while(l<=r){
mid=(l+r)/2;
if(candles[mid]>start){
r=mid-1;
}else if(candles[mid]<start){
l=mid+1;
}else{
l=mid;
break;
}
}
left=l;
l=0;
r=candles.size()-1;
while(l<=r){
mid=(l+r)/2;
if(candles[mid]>end){
r=mid-1;
}else if(candles[mid]<end){
l=mid+1;
}else{
r=mid;
break;
}
}
right=r;
cout<<left<<" "<<right<<endl;
if(right>=0&&left<candles.size())
res[i]=max(0,presum[candles[right]]-presum[candles[left]]);
}
return res;
}
};