2055. 蜡烛之间的盘子

这是我参与11月更文挑战的第9天，活动详情查看：2021最后一次更文挑战

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 <= $10^5$
s 只包含字符 '*' 和 '|' 。
1 <= queries.length <= $10^5$
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;
    }
};