LeetCode——蜡烛间的盘子（前缀和+预处理）

题目

每日一题，今天抽到的题目是蜡烛间的盘子，题目要求如下：

给你一个长桌子，桌子上盘子和蜡烛排成一列。给你一个下标从 0 开始的字符串 s ，它只包含字符 '*' 和 '|' ，其中 '*' 表示一个 盘子 ，'|' 表示一支 蜡烛 。

同时给你一个下标从 0 开始的二维整数数组 queries ，其中 queries[i] = [lefti, righti] 表示 子字符串 s[lefti...righti] （包含左右端点的字符）。对于每个查询，你需要找到 子字符串中 在 两支蜡烛之间 的盘子的 数目 。如果一个盘子在 子字符串中 左边和右边 都 至少有一支蜡烛，那么这个盘子满足在 两支蜡烛之间 。

• 比方说，s = "||**||**|*" ，查询 [3, 8] ，表示的是子字符串 "*||**|" 。子字符串中在两支蜡烛之间的盘子数目为 2 ，子字符串中右边两个盘子在它们左边和右边 都 至少有一支蜡烛。

请你返回一个整数数组 answer ，其中 answer[i] 是第 i 个查询的答案。

示例：

输入：s = "**|**|***|", queries = [[2,5],[5,9]]
输出：[2,3]
解释：
- queries[0] 有两个盘子在蜡烛之间。
- queries[1] 有三个盘子在蜡烛之间。

题目解析

刚开始看到这道题的第一想法是想再写一个方法，用于寻找每个区间中的盘子数量，在每次的寻找中找到该区间内所有蜡烛的位置，再通过蜡烛所在的位置相减得到该区间内的盘子数量。通过遍历整个二维数组，找到每个区间内的盘子数之和。开始这么写的就感觉这么写会超时，结果写完后果然超时了，这个方法虽然可以得到正确结果，但是太费时间了。

因此后面我换了另一种方法，转换了以下思路。既然每次都去寻找太浪费时间，那么我可以做一个预处理，提前找到所有蜡烛的位置，并记录在数组中。

同理，我也可以用一个数组来表示盘子的数量，具体方法为用一个presum[i]数组表示从开始到i个位置为止一共有多少个盘子，这样我们想要计算某区间中盘子的数量，先找到该区间内最右侧的蜡烛在哪里和最左侧的盘子在哪里，比如此时最右侧的蜡烛在y处，最左侧的盘子再x处，那么就可以直接使用presum[y]-presum[x]就可以得到该区间内盘子的总数量。

盘子的数量很好计算，直接遍历字符串，看到*让盘子数量加一即可。我认为此题的难点在于如何找到某区间内最左侧的蜡烛和最右侧的蜡烛。前面说了我想用数组来记录蜡烛的位置，但是如果只用一个数组来记录某位置上最近的蜡烛的位置是不够的，因为只是从一遍开始遍历，所以一个数组只能记录下最左侧蜡烛的位置和最右侧蜡烛的位置。

就拿示例中的字符串来说，如果只是用一个数组从左遍历来记录，我我们会得到以下数组

[-1,-1,2,2,2,5,5,5,5,9]

记录的只是每个位置最左侧的蜡烛位置。

因此我们要使用两个数组来记录每个位置上蜡烛的位置，一个从左遍历，一个从右遍历，这样我们就可以成功拿到某区间内最左侧蜡烛的位置和最右侧蜡烛的位置。从而通过两个位置上前面盘子的总数量相减，计算出区间内的盘子数量，最后放入数组中，完成该题。

代码

class Solution {
    public int[] platesBetweenCandles(String s, int[][] queries) {
        int n = s.length();
        int[] preSum = new int[n];
        for (int i = 0, sum = 0; i < n; i++) {
            if (s.charAt(i) == '*') {
                sum++;
            }
            preSum[i] = sum;
        }
        int[] left = new int[n];
        for (int i = 0, l = -1; i < n; i++) {
            if (s.charAt(i) == '|') {
                l = i;
            }
            left[i] = l;
        }
        int[] right = new int[n];
        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 i = 0; i < queries.length; i++) {
            int[] query = queries[i];
            int x = right[query[0]], y = left[query[1]];
            ans[i] = x == -1 || y == -1 || x >= y ? 0 : preSum[y] - preSum[x];
        }
        return ans;
    }
}