0
点赞
收藏
分享

微信扫一扫

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;
    }
};
举报

相关推荐

0 条评论