给定一个字符串 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”字符。
题解:
00110011
可以看作-2 2 -2 2
负数代表0,每次取相邻的最小的绝对值即可
char[] chs = s.toCharArray();
int res = 0;
int cur = chs[0] - '0';
int count = 1;
int pre = 0;
for (int i = 1; i < chs.length; i ++){
if (chs[i] - '0' == cur){
count ++;
if (count <= pre)
res ++;
}else {
pre = count;
count = 1;
cur = chs[i] - '0';
res ++;
}
}
return res;
或者更容易理解的
class Solution {
public int countBinarySubstrings(String s) {
if (s == null || s.length() == 0)
return 0;
int len = s.length();
int[] sum = new int[len];
int d = -1;
for (int i = 0; i < len; i++) {
int w;
if (s.charAt(i) == '0')
w = -1;
else
w = 1;
if (i == 0 || s.charAt(i) != s.charAt(i - 1)) {
sum[++d] = w;
}
else {
sum[d]+=w;
}
}
int num = 0;
for (int i = 1; i <= d; i++) {
num += Math.min(Math.abs(sum[i]),Math.abs(sum[i-1]));
}
return num;
}
}