- 计数二进制子串
给定一个字符串 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”字符。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/count-binary-substrings
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
- 考虑存一个数组
a[maxn]
,其中每个元素a[i]
表示第i
个连续串的个数
- 于是
a[maxn]
数组相邻
两个元素的最小值
的和就是答案 - 即 a n s = ∑ i = 2 m a x n ( m i n ( a [ i ] , a [ i − 1 ] ) ) ans = \sum_{i=2}^{maxn}{(min(a[i],a[i-1]))} ans=∑i=2maxn(min(a[i],a[i−1]))
- 最后代码可以把
a[maxn]
数组优化掉
#define debug
#ifdef debug
#include <time.h>
#endif
#include <iostream>
#include <algorithm>
#include <vector>
#include <string.h>
#include <map>
#include <set>
#include <stack>
#include <queue>
#include <math.h>
#define MAXN ((int)1e5+7)
#define ll long long int
#define INF (0x7f7f7f7f)
#define fori(lef, rig) for(int i=lef; i<=rig; i++)
#define forj(lef, rig) for(int j=lef; j<=rig; j++)
#define fork(lef, rig) for(int k=lef; k<=rig; k++)
#define QAQ (0)
using namespace std;
#define show(x...) \
do { \
cout << "\033[31;1m " << #x << " -> "; \
err(x); \
} while (0)
void err() { cout << "\033[39;0m" << endl; }
template<typename T, typename... A>
void err(T a, A... x) { cout << a << ' '; err(x...); }
namespace FastIO{
char print_f[105];
void read() {}
void print() { putchar('\n'); }
template <typename T, typename... T2>
inline void read(T &x, T2 &... oth) {
x = 0;
char ch = getchar();
ll f = 1;
while (!isdigit(ch)) {
if (ch == '-') f *= -1;
ch = getchar();
}
while (isdigit(ch)) {
x = x * 10 + ch - 48;
ch = getchar();
}
x *= f;
read(oth...);
}
template <typename T, typename... T2>
inline void print(T x, T2... oth) {
ll p3=-1;
if(x<0) putchar('-'), x=-x;
do{
print_f[++p3] = x%10 + 48;
} while(x/=10);
while(p3>=0) putchar(print_f[p3--]);
putchar(' ');
print(oth...);
}
} // namespace FastIO
using FastIO::print;
using FastIO::read;
int n, m, Q, K;
int a[MAXN], sz;
class Solution {
public:
int countBinarySubstrings(string s) {
int n = s.length();
if(!n) return 0;
sz = 0;
int cnt = 0, lst = s[0];
for(int i=0; lst; ) { //统计a[]数组
if(i == n || s[i] != lst) {
a[++sz] = cnt;
cnt = 0;
lst = s[i];
} else
cnt ++, i ++;
}
int sum = 0;
for(int i=2; i<=sz; i++) //对于a[]里的每个相邻对的最小值累加就是答案
sum += min(a[i], a[i-1]);
return sum;
}
};
#ifdef debug
signed main() {
Solution s;
cout << s.countBinarySubstrings("001100") << endl;
return 0;
}
#endif