leetcode 696. 计数二进制子串
题目详情
题目链接
给定一个字符串 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”字符。
我的代码
class Solution {
public:
int countBinarySubstrings(string s) {
int zero = 0, one = 0;
int n = s.size();
int res = 0;
int i = 0;
while (i < n) {
int loc = i;
if (s[i] == '0') {
while ((loc < n) && (s[loc] == '0')) {
++zero, ++loc;
}
i = loc;
res += min(one, zero);
one = 0;
} else {
while ((loc < n) && (s[loc] == '1')) {
++one, ++loc;
}
i = loc;
res += min(one, zero);
zero = 0;
}
}
return res;
}
};
我的成绩
执行结果:通过
执行用时:40 ms, 在所有 C++ 提交中击败了71.71%的用户
内存消耗:10.5 MB, 在所有 C++ 提交中击败了72.82%的用户
一些想法
本道题是一道简单题,我们要做的就是尽可能的减少重复计算的次数,只需进行一次遍历。
执行用时为 24 ms 的范例
class Solution {
public:
int countBinarySubstrings(string s) {
int n = s.size();
int dp[n];
dp[0] = 0;
int zero = 0;
int one = 0;
char now = s[0];
if(now == '0') zero++;
else one++;
for(int i = 1; i < n; i++){
if(s[i] == now){
if(now == '0'){
zero++;
}
else one++;
}
else{
if(now =='0'){
now = '1';
one = 1;
}
else{
now ='0';
zero = 1;
}
}
if(now == '0' && zero <= one) dp[i] = dp[i-1] + 1;
else if(now =='1' && one <= zero) dp[i] = dp[i-1] + 1;
else dp[i] = dp[i-1];
}
return dp[n-1];
}
};
思考
参见官方解答