【LeetCode每日一题】

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

题解:
方法一:这道题可以使用前缀和的方案,不知道为什么写了特别特别久,不过最后还是调出来了。第一种方法是仅仅用前缀和预处理一个sum数组,用来存储从开头到当前位置的蜡烛的数量,并用list数组存储下盘子的位置。在查询的时候使用二分查找。
class Solution {
public:
    vector<int> platesBetweenCandles(string s, vector<vector<int>>& queries) {
        int n = s.size();
        vector<int> sum(n + 1, 0);
        vector<int> list;
        for(int i = 0; i < n; i++){
            sum[i + 1] = sum[i] + (s[i] == '*' ? 1 : 0);
            if(s[i] == '|') list.push_back(i);
        }
        //cout<<"size:"<<list.size()<<endl;
        vector<int> ans(queries.size());
        if(list.size() == 0) return ans;
        for(int c = 0; c < queries.size(); c++){
            int a = -1, b = -1;
            int l = 0, r = list.size() - 1;
            while(l < r){
                int m = (l + r) / 2;
                if(list[m] >= queries[c][0]) r = m;
                else l = m + 1;
            }
            a = list[l];
            l = 0, r = list.size() - 1;
            while(l < r){
                int m = (l + r + 1) / 2;
                if(list[m] <= queries[c][1]) l = m;
                else r = m - 1;
            }
            b = list[r];
            //cout<<a<<" "<<b<<endl;
            if(a != -1 && b != -1 && a <= b) ans[c] = sum[b + 1] - sum[a];
        }
        return ans;
    }
};
方法二:
第二种方法是直接用前缀和预处理出蜡烛的数量,以及当前位置左/右边的盘子的下标。这样直接就可以求解了。
class Solution {
public:
    vector<int> platesBetweenCandles(string s, vector<vector<int>>& queries) {
        int n = s.length();
        vector<int> preSum(n);
        for (int i = 0, sum = 0; i < n; i++) {
            if (s[i] == '*') {
                sum++;
            }
            preSum[i] = sum;
        }
        vector<int> left(n);
        for (int i = 0, l = -1; i < n; i++) {
            if (s[i] == '|') {
                l = i;
            }
            left[i] = l;
        }
        vector<int> right(n);
        for (int i = n - 1, r = -1; i >= 0; i--) {
            if (s[i] == '|') {
                r = i;
            }
            right[i] = r;
        }
        vector<int> ans;
        for (auto& query : queries) {
            int x = right[query[0]], y = left[query[1]];
            ans.push_back(x == -1 || y == -1 || x >= y ? 0 : preSum[y] - preSum[x]);
        }
        return ans;
    }
};
空间复杂度均为O(n),时间复杂度分别为O(n + qlogn)和O(n + q)
全部评论

相关推荐

头像
点赞 评论 收藏
转发
点赞 评论 收藏
转发
1 收藏 评论
分享
牛客网
牛客企业服务