2019牛客暑期多校训练营 第四场 K – number (思维+前缀和)
程序员文章站
2022-04-02 22:09:44
...
题目链接:
题意:给你一个字符串让你求出有多少个子串mod300等于0
是300的倍数 肯定是3的倍数也是100的倍数
所以处理的时候每遇到00就要把前面3的倍数的区间加上
区间内的数字加和是%3等于0就是3的倍数
[l,r]的区间和 可以等于 sum[r]-sum[l-1]的区间和
(sum[r]-sum[l-1])%3 = 0 ---> sum[r]%3=sum[l-1]%3
所以当前前缀和出现一次之后就加上前面这个前缀和出现的次数,就是%3等于零的区间个数
#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int maxn = 1e6+6;
char s[maxn];
ll sum[maxn];
ll num[3];
ll ans;
int main(){
while(~scanf("%s",s+1)) {
int len = strlen(s+1);
sum[0]=0;ans=0;
num[0]=num[1]=num[2]=0;
for(int i=1;i<=len;i++) sum[i]=(sum[i-1]+s[i]-'0')%3;
for(int i=1;i<=len;i++) {
if(s[i]=='0' && s[i-1]=='0') {
ans+=num[sum[i-1]];
}
if(s[i]=='0')ans++;
num[sum[i-1]]++;
}
printf("%lld\n",ans);
}
return 0;
}