*python3__leecode/2055. 蜡烛之间的盘子

一、刷题内容

原题链接

https://leetcode-cn.com/problems/plates-between-candles/

内容描述

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

import bisect

class Solution(object):
    def platesBetweenCandles(self, s, queries):
        #记录蜡烛的位置
        candle = []
        for i, n in enumerate(s):
            if n == '|': 
                candle.append(i)

        ans = []
        for q in queries:
            pre, post = q[0], q[1]
            pre_p  = bisect.bisect_left(candle, pre)
            post_p = bisect.bisect_left(candle, post)
            #二分查找左闭右开，右边需要判断一下是否相等
            if post_p == len(candle) or candle[post_p] > post:
                post_p -= 1
            if post_p <= pre_p: #只有一根蜡烛或者没蜡烛
                ans.append(0)
                continue
            ans.append(candle[post_p] - candle[pre_p] - (post_p - pre_p))

        return ans