22-03-08 蜡烛之间的盘子

蜡烛之间的盘子

2055. 蜡烛之间的盘子 难度[中等]

---

给你一个长桌子，桌子上盘子和蜡烛排成一列。给你一个下标从 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
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

---

解法一：暴力超时

class Solution {
    public int[] platesBetweenCandles(String s, int[][] queries) {
        int n = queries.length;
        int[] ans = new int[n];
        int idx = 0;
        for (int[] q :queries) {
            int l = q[0];
            int r = q[1];
            while (l < r) {
                if ('|' != s.charAt(l)) {
                    l++;
                }
                if ('|' != s.charAt(r)) {
                    r--;
                }
                if ('|' == s.charAt(l) && '|' == s.charAt(r)) {
                    break;
                }
            }
            for (int i = l+1; i < r; i++) {
                if ('*' == s.charAt(i)){
                    ans[idx] += 1;
                }
            }
            idx++;
        }
        return ans;
    }
}

解法二：前缀和 + 二分

class Solution {
    public int[] platesBetweenCandles(String s, int[][] queries) {
        char[] cs =  s.toCharArray();
        int n = queries.length;
        int m = cs.length;
        // 答案 与 前缀和
        int[] ans = new int[n], sum = new int[m+1];
        // '|' 的下标
        List<Integer> list = new ArrayList();
        for (int i = 0; i < m; i++) {
            if (cs[i] == '|') list.add(i);
            sum[i+1] = sum[i] + (cs[i] == '*' ? 1 : 0);
        }
        if (list.size() == 0) return ans;
        for (int i = 0; i < n; i++) {
            int[] q = queries[i];
            int l = 0, r = list.size()-1;
            //离所给范围中下边界最近的 '|' 的下标
            int L = -1, R = -1;
            //找到左边 '|' 的下标
            while (l < r) {
                int mid = l+r >> 1;
                if (list.get(mid) >= q[0]) r = mid;
                else l = mid+1;
            }
            if (list.get(r) >= q[0]) L = list.get(r);
            else continue;
            //找到右边 '|' 的下标
            l = 0;
            r = list.size()-1;
            while (l < r) {
                int mid = l+r+1 >> 1;
                if (list.get(mid) <= q[1]) l = mid;
                else r = mid-1;
            }
            if (list.get(r) <= q[1]) R = list.get(r);
            else continue;
            if (L <= R) ans[i] = sum[R+1] - sum[L];
        }
        return ans;
    }
}

参考

---

此文章创于本人学习时的记录，如有错误或更优解还请指出