Leetcode.2055 蜡烛之间的盘子

题目链接

Leetcode.2055 蜡烛之间的盘子 rating : 1819

题目描述

给你一个长桌子，桌子上盘子和蜡烛排成一列。给你一个下标从 $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\le s.length\le 10^5$
• $s$ 只包含字符 '*' 和 '|' 。
• $1\le queries.length\le 10^5$
• $queries[i].length==2$
• $0\le lefti\le righti<s.length$

解法：前缀和

用 $presum$ 记录蜡烛的前缀和数量。例如 $presum[r]-presum[l-1]$ 就是区间 $[l,r]$ 内的盘子数量。

用 $left$ 记录每一个位置左边蜡烛的位置。例如 $left[i]$ 就是位置 $i$ 左边最近的蜡烛的位置，没有的话默认为 $-1$。

用 $right$ 记录每一个位置右边蜡烛的位置。例如 $right[i]$ 就是位置 $i$ 右边最近的蜡烛的位置，没有的话默认为 $n$。

对于每一个查询 $queries[i]=[a,b]$，设 $l=right[a],r=left[b]$。$[l,r]$ 就是我们要计算盘子的有效范围，如果$l\ge r$，说明区间 $[a,b]$ 内没有有效的盘子；否则，区间 $[a,b]$ 之内的有效盘子数量为 ： $presum[r]-presum[l]$。

时间复杂度： $O(n)$

C++代码：

class Solution {
public:
    vector<int> platesBetweenCandles(string s, vector<vector<int>>& queries) {
        int n = s.size();
        vector<int> presum(n);
        for(int i = 0,sum = 0;i < n;i++){
            if(s[i] == '*') sum++;
            presum[i] = sum;
        }

        vector<int> left(n , -1);
        for(int i = 0,pre = -1;i < n;i++){
            if(s[i] == '|') pre = i;
            left[i] = pre;
        }

        vector<int> right(n,n);
        for(int i = n - 1,pre = n;i >= 0;--i){
            if(s[i] == '|') pre = i;
            right[i] = pre;
        }

        int m = queries.size();
        vector<int> ans(m);

        for(int i = 0;i < m;i++){
            int a = queries[i][0] , b = queries[i][1];
            int l = right[a] , r = left[b];
            if(l >= r) continue;
            ans[i] = presum[r] - presum[l];
        }
        

        return ans;
    }
};