给定一个字符串 s,统计并返回具有相同数量 0 和 1 的非空(连续)子字符串的数量,并且这些子字符串中的所有 0 和所有 1 都是成组连续的。
重复出现(不同位置)的子串也要统计它们出现的次数。
示例 1:
输入:s = "00110011"
输出:6
解释:6 个子串满足具有相同数量的连续 1 和 0 :"0011"、"01"、"1100"、"10"、"0011" 和 "01" 。
注意,一些重复出现的子串(不同位置)要统计它们出现的次数。
另外,"00110011" 不是有效的子串,因为所有的 0(还有 1 )没有组合在一起。
示例 2:
输入:s = "10101"
输出:4
解释:有 4 个子串:"10"、"01"、"10"、"01" ,具有相同数量的连续 1 和 0 。
提示:
1 <= s.length <= 105s[i]为'0'或'1'
思路:
问题实际在问:对于每一对相邻的连续相同字符段——比如一段0和相邻的一段1——计算它们能形成多少有效子串。所以,对于长度为 len0 的0段和长度为 len1 的1段,可以形成 min(len0, len1) 个有效子串。那么第一步就是将字符串按连续相同字符分组,记录每组长度,第二步则是对于相邻的两段,累加 min(前一段, 后一段) 即可。
AC代码:
class Solution {
public:
int countBinarySubstrings(string s) {
if (s.empty()) return 0;
vector<int> groups;
int count = 1;
for (int i = 1; i < s.size(); i++) {
if (s[i] == s[i-1]) {
count++;
} else {
groups.push_back(count);
count = 1;
}
}
groups.push_back(count);
int result = 0;
for (int i = 1; i < groups.size(); i++) {
result += min(groups[i-1], groups[i]);
}
return result;
}
};Code language: C++ (cpp)