给你一个长桌子,桌子上盘子和蜡烛排成一列。给你一个下标从 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 个盘子在蜡烛之间。
- 另一个查询没有盘子在蜡烛之间。
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/plates-between-candles
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题思路
参照leetcode官方方法,利用前缀和的思想。透过题目可以发现,实质上是在求最左和最右两个蜡烛两个蜡烛之间有多少个碟子。
1、首先利用一个数组记录0–i之间有多少个碟子;
2、记录0–i之间最右边的’|‘的坐标,一开始没有r设成-1;
3、记录s.length()-1到i之间最左边的’|‘的坐标,一开始没有r设成-1;
4、获得最左的"|“的坐标,获得最右的”|"的坐标;
5、最右的’|‘之前有多少碟子(preSum)减去最左的’|'之前有多少碟子(preSum)就得到两个"|"之间的碟子的数量。
class Solution {
public int[] platesBetweenCandles(String s, int[][] queries) {
int cnt[] = new int[queries.length];
int preSum[] = new int[s.length()];
int sum = 0;
//0--i之间有多少个‘*’
for (int i = 0; i < s.length(); i++) {
if (s.charAt(i) == '*'){
sum++;
}
preSum[i] = sum;
}
//记录0--i之间最右边的'|'的坐标,一开始没有r设成-1
int [] left = new int[s.length()];
for (int i = 0, l = -1; i < s.length(); i++) {
if (s.charAt(i) == '|'){
l = i;
}
left[i] = l;
}
//记录s.length()-1到i之间最左边的'|'的坐标,一开始没有r设成-1
int [] right = new int[s.length()];
for (int i = s.length() - 1, r = -1; i >= 0; i--) {
if (s.charAt(i) == '|'){
r = i;
}
right[i] = r;
}
for(int i = 0; i < queries.length; i++){
int start = queries[i][0];
int end = queries[i][1];
int x = right[start];//最左的"|"
int y = left[end];//最右的"|"
if (x == -1 || y == -1 || x >= y){
cnt[i] = 0;
}else {
//最右的'|'之前有多少"*"减去最左的'|'之前有多少"*"就得到两个"|"之间的"*"的数量
cnt[i] = preSum[y] - preSum[x];
}
}
return cnt;
}
}