欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

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;
} 

 

相关标签: 思维