【LeetCode】2055. 蜡烛之间的盘子

题目

2055. 蜡烛之间的盘子

难度中等59收藏分享切换为英文接收动态反馈

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

题解1

思路

• 暴力遍历每一种情况
• TLE

代码

class Solution:
    def platesBetweenCandles(self, s: str, queries: List[List[int]]) -> List[int]:
        res = []
        for start,stop in queries:
            count = tempCount = 0
            i=j=start
            while s[i]!="|" and i<= stop:
                i+=1
                j+=1
            while j <= stop:
                if s[j] == "*":
                    tempCount += 1
                elif s[j] == "|":
                    count += tempCount
                    tempCount = 0
                j+=1
            res.append(count)
        return res

复杂度

• 时间复杂度：$O(n^2)$
• 空间复杂度：$O(n)$

题解2

思路

• 计算每一点的盘子数，以及每一点最近的右侧、左侧的蜡烛位置
• 遍历queries中的每一个子列，其中盘子数量等于其子列中两个距离端点最近的蜡烛之间的盘子数

代码

class Solution:
    def platesBetweenCandles(self, s: str, queries: List[List[int]]) -> List[int]:
        left,right,preSum = [0]*len(s),[0]*len(s),[0]*len(s)
        leftCandle,rightCandle, count = -1,len(s),0
        res = []
        for i,c in enumerate(s):
            if c =="*":
                count += 1
            else:
                leftCandle = i
            preSum[i] = count
            left[i] = leftCandle
        for i in range(len(s)-1,-1,-1):
            if s[i] == "|":
                rightCandle = i
            right[i] = rightCandle
        for start,stop in queries:
            start,stop = right[start],left[stop]
            if start>=0 and stop>=0 and stop < len(s) and start<stop:
                res.append(preSum[stop] - preSum[start])
            else:
                res.append(0)
        return res
        
            

复杂度

• 时间复杂度：$O(n+q)$
• 空间复杂度：$O(n)$