Question

2055. 蜡烛之间的盘子

English Version

题目描述

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

解法

方法一

class Solution:
  def platesBetweenCandles(self, s: str, queries: List[List[int]]) -> List[int]:
    n = len(s)
    presum = [0] * (n + 1)
    for i, c in enumerate(s):
      presum[i + 1] = presum[i] + (c == '*')

    left, right = [0] * n, [0] * n
    l = r = -1
    for i, c in enumerate(s):
      if c == '|':
        l = i
      left[i] = l
    for i in range(n - 1, -1, -1):
      if s[i] == '|':
        r = i
      right[i] = r

    ans = [0] * len(queries)
    for k, (l, r) in enumerate(queries):
      i, j = right[l], left[r]
      if i >= 0 and j >= 0 and i < j:
        ans[k] = presum[j] - presum[i + 1]
    return ans

class Solution {
  public int[] platesBetweenCandles(String s, int[][] queries) {
    int n = s.length();
    int[] presum = new int[n + 1];
    for (int i = 0; i < n; ++i) {
      presum[i + 1] = presum[i] + (s.charAt(i) == '*' ? 1 : 0);
    }
    int[] left = new int[n];
    int[] right = new int[n];
    for (int i = 0, l = -1; i < n; ++i) {
      if (s.charAt(i) == '|') {
        l = i;
      }
      left[i] = l;
    }
    for (int i = n - 1, r = -1; i >= 0; --i) {
      if (s.charAt(i) == '|') {
        r = i;
      }
      right[i] = r;
    }
    int[] ans = new int[queries.length];
    for (int k = 0; k < queries.length; ++k) {
      int i = right[queries[k][0]];
      int j = left[queries[k][1]];
      if (i >= 0 && j >= 0 && i < j) {
        ans[k] = presum[j] - presum[i + 1];
      }
    }
    return ans;
  }
}

class Solution {
public:
  vector<int> platesBetweenCandles(string s, vector<vector<int>>& queries) {
    int n = s.size();
    vector<int> presum(n + 1);
    for (int i = 0; i < n; ++i) presum[i + 1] = presum[i] + (s[i] == '*');
    vector<int> left(n);
    vector<int> right(n);
    for (int i = 0, l = -1; i < n; ++i) {
      if (s[i] == '|') l = i;
      left[i] = l;
    }
    for (int i = n - 1, r = -1; i >= 0; --i) {
      if (s[i] == '|') r = i;
      right[i] = r;
    }
    vector<int> ans(queries.size());
    for (int k = 0; k < queries.size(); ++k) {
      int i = right[queries[k][0]];
      int j = left[queries[k][1]];
      if (i >= 0 && j >= 0 && i < j) ans[k] = presum[j] - presum[i + 1];
    }
    return ans;
  }
};

func platesBetweenCandles(s string, queries [][]int) []int {
  n := len(s)
  presum := make([]int, n+1)
  for i := range s {
    if s[i] == '*' {
      presum[i+1] = 1
    }
    presum[i+1] += presum[i]
  }
  left, right := make([]int, n), make([]int, n)
  for i, l := 0, -1; i < n; i++ {
    if s[i] == '|' {
      l = i
    }
    left[i] = l
  }
  for i, r := n-1, -1; i >= 0; i-- {
    if s[i] == '|' {
      r = i
    }
    right[i] = r
  }
  ans := make([]int, len(queries))
  for k, q := range queries {
    i, j := right[q[0]], left[q[1]]
    if i >= 0 && j >= 0 && i < j {
      ans[k] = presum[j] - presum[i+1]
    }
  }
  return ans
}