给你一个长桌子,桌子上盘子和蜡烛排成一列。给你一个下标从 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
解题思路:
解题关键点在于需要知道每个点左右木板的位置和右边木板的位置。例如对于示例1中的[2, 5],就需要知道2右边木板的位置,以及5左边木板的位置。得到位置后如果知道2-5之间的蜡烛数量呢?这里考虑使用前缀树思想,[2- 5] = [0 - 5] - [0 - 2]。2,5表示木板的位置。通过一次遍历就能够得到所有点到0这个点中间蜡烛的数量。同时考虑边界情况,如果第一个点左边没有木板或者第二个点右边没有木板则他们之前没有蜡烛。代码如下:
class Solution {
public:
vector<int> platesBetweenCandles(string s, vector<vector<int>>& queries) {
//记录每个点左边的木板,记录木板的下一个木板,记录该木板后的长度
int n = s.size(), sum = 0, l = -1, r = -1 ;
vector<int>left(n,-1),right(n, -1),count(n, 0),res;
for (int i = 0; i < n; ++i) {
if (s[i] == '*') {
sum++;
}
count[i] = sum;
if (s[i] == '|') {
l = i;
}
if (s[n - i - 1] == '|') {
r = n - i - 1;
}
left[i] = l;
right[n - i - 1] = r;
}
for (auto item : queries) {
l = right[item[0]];
r = left[item[1]];
if ( l < r && l != -1 && r != -1)
res.push_back(count[r] - count[l]);
else
res.push_back(0);
}
return res;
}
};