LeetCode 2055 蜡烛之间的盘子[预处理 前缀和] HERODING的LeetCode之路

解题思路：
刚开始看到这道题目的时候，脑海里就想到预处理的方法，把每个盘子左右两边的蜡烛位置标记下来，在查询的时候，只要对范围中的盘子进行判断，如果盘子的左右蜡烛都在范围内，那么就记录下来，这样就有了如下的代码：

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

很遗憾，最终该代码超时了，问题很大部分来自于查询部分，能不能给定范围就能直接判断范围中的盘子而不是直接查找呢？这就是前缀和的作用，通过前缀和记录之前盘子的总数，在查询时，只要找到范围中的边界蜡烛，通过前缀和相减就得到了蜡烛之间盘子的个数，代码如下：

class Solution {
public:
    vector<int> platesBetweenCandles(string s, vector<vector<int>>& queries) {
        int n = s.size();
        vector<int> left(n);
        vector<int> right(n);
        vector<int> preSum(n);
        vector<int> ans;
        int sum = 0;
        // 计算前缀和
        for(int i = 0; i < n; i ++) {
            if(s[i] == '*') {
                sum ++;
            }
            preSum[i] = sum;
        }
        int cur = -1;
        // 记录左边蜡烛的位置
        for(int i = 0; i < n; i ++) {
            if(s[i] == '|') {
                cur = i;
            } 
            left[i] = cur;
        }
        cur = n;
        // 记录右边盘子的位置
        for(int i = n - 1; i >= 0; i --) {
            if(s[i] == '|') {
                cur = i;
            }
            right[i] = cur;
        }
        for(auto& q : queries) {
            int l = right[q[0]], r = left[q[1]];
            int count = (l < n && r > -1 && l < r) ? preSum[r] - preSum[l] : 0;
            ans.push_back(count);
        }
        return ans;
    }
};