一、题目
给你一个长桌子,桌子上盘子和蜡烛排成一列。给你一个下标从 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
第一解法
获取到开始结束区间后,直接在相应的字符字串中进行遍历
class Solution {
public:
vector<int> platesBetweenCandles(string s, vector<vector<int>>& queries) {
vector<int> answer;
int start = 0, end = 0;
int length = s.length();
for (int i = 0; i < queries.size(); ++i)
{
start = queries[i][0];
end = queries[i][1];
if (start > length || end > length)
answer.push_back(0);
else
{
int count = 0;
int seg_count = 0;
int tmp_count = 0;
bool is_between = false;
for (int j = start; j <= end; ++j)
{
if (s[j] == '|' && is_between)
{
is_between = false;
seg_count++;
count += tmp_count;
tmp_count = 0;
}
if (s[j] == '*' && is_between)
tmp_count++;
if (j - 1 >= start)
{
if (s[j] == '*' && s[j - 1] == '|' && !is_between)
{
tmp_count++;
is_between = true;
}
}
}
if (seg_count == 0)
{
count = 0;
answer.push_back(count);
}
else
{
answer.push_back(count);
}
}
}
return answer;
}
};
结果: 超时
第二解法
class Solution {
public:
vector<int> platesBetweenCandles(string s, vector<vector<int>>& queries) {
int length = s.length();
//计算前缀和
vector<int> presum(length);
for (int i = 0, sum = 0; i < length; ++i)
{
if (s[i] == '*')
{
sum++;
}
presum[i] = sum;
}
//从左边计算蜡烛位置
vector<int> llight(length);
for (int i = 0, l = -1; i < length; ++i)
{
if (s[i] == '|')
{
l = i;
}
llight[i] = l;
}
//从右边计算蜡烛位置
vector<int> rlight(length);
for (int i = length - 1, r = -1; i >= 0; --i)
{
if (s[i] == '|')
{
r = i;
}
rlight[i] = r;
}
//返回结果
vector<int> answer;
for (auto& query : queries)
{
int x = rlight[query[0]], y = llight[query[1]];
answer.push_back(x == -1 || y == -1 || x >= y ? 0 : presum[y] - presum[x]);
}
return answer;
}
};
因为llight数组的含义时i的左边第一根蜡烛位置,rlight数组的含义为i右边第一根蜡烛的位置,列如区间[2,5],2需要找其右边第一根蜡烛的位置,即rlight[2],5需要找其左边第一根蜡烛的位置,即rlight[5]。最后得其前缀和得插值,此方法时间复杂度为O(1);