给定一个字符串 s,计算具有相同数量0和1的非空(连续)子字符串的数量,并且这些子字符串中的所有0和所有1都是组合在一起的。
重复出现的子串要计算它们出现的次数。
示例 1 :
输入: "00110011"
输出: 6
解释: 有6个子串具有相同数量的连续1和0:“0011”,“01”,“1100”,“10”,“0011” 和 “01”。
请注意,一些重复出现的子串要计算它们出现的次数。
另外,“00110011”不是有效的子串,因为所有的0(和1)没有组合在一起。
示例 2 :
输入: "10101"
输出: 4
解释: 有4个子串:“10”,“01”,“10”,“01”,它们具有相同数量的连续1和0。
注意:
s.length 在1到50,000之间。
s 只包含“0”或“1”字符。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/count-binary-substrings
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解决方案:
我们发现数据范围在5*1e4,暴力求解一定会tle。
题目要求获得相同数量的连0连1子串,因此我们可以使用一临时变量preCount存储之前那个连0/连1串的长度,curCount存储当前连0/连1串的长度。可以只遍历一次即可求解。
例如对于如下案例
"00111"
遍历到最后一个位置时,preCount = 2, curCount = 3。
此外我们发现对于每次能够组成满足条件的子串数目为min(preCount, curCount)
"00111"
01 0011
代码如下:
class Solution {
public int countBinarySubstrings(String s) {
// init
int preCount = 0;
int curCount = 1;
char cur = s.charAt(0);
int ans = 0;
for(int i = 1; i < s.length(); i++){
if(s.charAt(i) == cur){
curCount++;
}else{
ans += Math.min(preCount, curCount);
preCount = curCount;
curCount = 1;
cur = s.charAt(i);
}
}
ans += Math.min(preCount, curCount);
return ans;
}
}