LeetCode 2055. 蜡烛之间的盘子（前缀和）

Michael阿明

发布于 2022-01-07 11:06:32

4390

文章目录
1. 题目
2. 解题

1. 题目

给你一个长桌子，桌子上盘子和蜡烛排成一列。给你一个下标从 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

来源：力扣（LeetCode）链接：https://leetcode-cn.com/problems/plates-between-candles 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2. 解题

求出每个位置左侧的最近的 | 的位置，同理右侧的也是一样
然后找到区间左端点右侧最近的 | 的位置，右端点左侧最近的 | 的位置
利用前缀和做差求解 * 的数量

class Solution {
public:
    vector<int> platesBetweenCandles(string s, vector<vector<int>>& queries) {
        int n = s.size(), p = -1;
        vector<int> left(n, INT_MIN), right(n, INT_MAX);
        for(int i = 0; i < n; ++i)
        {
            if(s[i] == '|')
                p = i;
            if(p != -1)
                left[i] = p;
        }
        p = -1;
        for(int i = n-1; i >= 0; --i)
        {
            if(s[i] == '|')
                p = i;
            if(p != -1)
                right[i] = p;
        }
        vector<int> presum(n, 0); // 计算*的前缀数量
        presum[0] = s[0]=='*'? 1 : 0;
        for(int i = 1; i < n; ++i)
        {
            if(s[i]=='*')
                presum[i] = presum[i-1]+1;
            else
                presum[i] = presum[i-1];
        }
        vector<int> ans(queries.size());
        for(int i = 0; i < queries.size(); ++i)
        {
            int l = right[queries[i][0]];
            int r = left[queries[i][1]];
            if(l < r)
               ans[i] = presum[r]-presum[l]; 
        }
        return ans;
    }
};

356 ms 138.7 MB C++ 2021/11/06 10:25

我的CSDN博客地址 https://michael.blog.csdn.net/

本文参与腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。

原始发表：2021/11/06 ，如有侵权请联系 cloudcommunity@tencent.com 删除

https