01序列 - 题目 - Daimayuan Online Judge
大致题意是字符串中只有0和1,找出有多少个子串含有k个1
依次记录每个1的下标,枚举每个1,运用乘法原理,比如01010100,找含有2个1的子串,先枚举第一个橙色的1,然后如果找到第2个橙色的1,从第一个橙色的1开始往左数(包括它自己),数尽所有的0(数到上一个1右边的0为止),乘从第二个橙色的1开始往右数(包括它自己),数尽所有的0(数到下一个1左边的0为止),即2*2=4,再枚举第二个橙色的1,同理2*3=6,加起来就是10
然后交上去代码是错误的,原因是这只适用于k不为0的情况,如果k为0,数的时候就要避开1,于是分情况讨论,k不为0的时候就是上面那种情况,当k为0的时候,也用乘法原理,枚举每一个0,比如0001,含有第一个0的子串个数就是1*3,含有第二个0的子串个数就是0*2,含有第三个0的子串个数就是0*1,加起来就行了
乘法原理具体解释见(3条消息) 每日一题:子串的最大差_m0_74087709的博客-CSDN博客
AC代码:
#include<iostream>
#include<algorithm>
#include<cstring>
#define int long long
using namespace std;
const int N = 1e6 + 10;
int a[N];
int cnt;
signed main()
{
int k;
string s;
cin >> k >> s;
for (int i = 0; i < s.size(); i++) {
if (s[i] == '1') {
cnt++;
a[cnt] = i;
}
}
int res = 0;
a[0] = -1;
if (k == 0) {
for (int i = 1; i <= cnt; i++) {
for (int j = a[i - 1] + 1; j <= a[i] - 1; j++) {
res += a[i] - j;
}
}
for (int j = a[cnt] + 1; j < s.size(); j++) {
res += s.size() - j;
}
}
else
for (int i = 1; i <= cnt; i++) {
if (i + k-1 <= cnt) {
if(i+k<=cnt)
res += (a[i] - a[i - 1]) * (a[i + k] - a[i + k - 1]);
else {
res+= (a[i] - a[i - 1]) * (s.size() - a[i + k - 1]);
}
}
}
cout << res << endl;
}