题目简介:
给定一个字符串 s
,计算具有相同数量0和1的非空(连续)子字符串的数量,并且这些子字符串中的所有0和所有1都是组合在一起的。
重复出现的子串要计算它们出现的次数。
示例 1 :
1 2 3 4 5 6 7
| 输入: "00110011" 输出: 6 解释: 有6个子串具有相同数量的连续1和0:“0011”,“01”,“1100”,“10”,“0011” 和 “01”。
请注意,一些重复出现的子串要计算它们出现的次数。
另外,“00110011”不是有效的子串,因为所有的0(和1)没有组合在一起。
|
示例 2 :
1 2 3
| 输入: "10101" 输出: 4 解释: 有4个子串:“10”,“01”,“10”,“01”,它们具有相同数量的连续1和0。
|
注意:
s.length
在1到50,000之间。
s
只包含“0”或“1”字符。
思路:
首先将字符串进行分组,把连续的1或0的数量存入数组,如字符串1100101
,可分为{2, 2, 1, 1, 1}
。
最后再逐个加上数组中相邻两项的较小值即可,即sum += min(count[i], count[i + 1])
。
代码如下:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30
| class Solution { public: int countBinarySubstrings(string s) {
vector<int> count;
int sum = 1;
for(int i = 1; i < s.size(); i++){
if(s[i] != s[i - 1]){
count.push_back(sum); sum = 0; }
sum++; }
count.push_back(sum);
sum = 0; for(int i = 0; i < count.size() - 1; i++){
sum += min(count[i], count[i + 1]); }
return sum; } };
|