2019牛客暑期多校训练营 第四场 K – number （思维+前缀和）

题目链接：

题意：给你一个字符串让你求出有多少个子串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
#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;
}