大打字家
题面∶· 小E今天要去参加信息学奥林匹克的竞赛,但是主办方没有告诉她自己账号的密码,而是给了一个很长的二进制数。
· 通过别的提示,小E猜测账号密码应该是该二进制数内均由1组成的子串的个数。
· 这个问题对小E来说太简单了。她想考考你,请问你能解答出来吗?
· 因为这个数字很大,所以我们只需要输出答案对20200821取模的结果就好了。
输入∶
·输入仅一行,为一个二进制数· 二进制数仅由0和1组成,且前面、中间均没有空格输出∶
· 一个整数,为对20200821取模的答案
数据范围∶
· 对于50%的数据,满足输入二进制数的长度不超过2000
· 对于100%的数据,满足输入二进制数的长度不超过500000
样例输入∶ 0100110
样例输出∶4
#include<cstdio>
#include<string>
#include<iostream>
using namespace std;
#define Mod 20200821
string s;
long long a[500007];//a[i]表示1+2+3+4+5+6+…+i-2+i-1+i
int main(){
cin>>s;//读入01串
int n=s.length();//01串的长度
long long ans=0;
a[0]=0;
for (int i = 1; i <= l; ++i){
a[i]=a[i-1]+i;
a[i]%=Mod;//先取模,后面加的时候答案不变
}
int cnt=0;
for (int i = 1; i <= l; ++i){
if(s[i]=='1'){
cnt++;
}else if(s[i]=='0'&&cnt>0){//如果前面有连续的为1的子串 (即cnt>0)且这位不是一,则计算cnt
ans+=a[cnt];
ans%=Mod;
cnt=0;
}
}
if(cnt>0){//如果最后以1结尾就特判
ans+=a[cnt];
ans%=Mod;
}
printf("%lld",ans);
return 0;
}
}