给定一个字符串 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) {
List<Integer> counts=new ArrayList<Integer>();
int ptr=0,n=s.length();
while(ptr<n) {
char c=s.charAt(ptr);
int count=0;
while(ptr<n&&s.charAt(ptr)==c) {
++ptr;
++count;
}
counts.add(count);
}//00110011先计算出counts中的值{2,2,2,2};
int ans=0;
for(int i=1;i<counts.size();++i) {
ans+=Math.min(counts.get(i),counts.get(i-1));
}
return ans;
}
}
这个实现的时间复杂度和空间复杂度都是O(n)。
对于某一个位置 i,其实我们只关心i−1 位置的counts 值是多少,所以可以用一个last 变量来维护当前位置的前一个位置,这样可以省去一个 counts 数组的空间。
class Solution {
public:
int countBinarySubstrings(string s) {
int ptr = 0, n = s.length(), last = 0, ans = 0;
while (ptr < n) {
char c = s.charAt(ptr);
int count = 0;
while (ptr < n && s.charAt(ptr) == c) {
++ptr;
++count;
}
ans += Math.min(count, last);
last = count;
}
return ans;
}
};
复杂度分析
时间复杂度:O(n)。
空间复杂度:O(1)。