给定一个字符串 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”字符。
C++
class Solution {
public:
int countBinarySubstrings(string s) {
//我们可以将字符串 s 按照 0 和 1 的连续段分组,存在 counts 数组中,
//例如 s = 00111011 ,可以得到这样的 counts 数组: counts = {2, 3, 1, 2}。
vector<int> counts;
int idx = 0, n = s.size();
while(idx < n){
char c = s[idx];
int count = 0;
while(idx < n && c == s[idx]){
idx++;
count++;
}
counts.push_back(count);
}
int res = 0;
//遍历counts数组,将每组相邻2个数的最小值全部累加就是答案。
for(int i = 1; i < counts.size(); i++){
res += min(counts[i], counts[i - 1]);
}
return res;
}
};
C++② (利用变量last维护前一个值)
class Solution {
public:
int countBinarySubstrings(string s) {
int res = 0, last = 0;
int idx = 0, n = s.size();
while(idx < n){
char c = s[idx];
int cnt = 0;
while(idx < n && c == s[idx]){
idx++;
cnt++;
}
res += min(last, cnt);
last = cnt;
}
return res;
}
};
参考:计数二进制子串