ABC164 D.Multiple of 2019

D.Multiple of 2019

Question

给一个字符串S，求有多少个子串在十进制下为2019的倍数。

Solution

前置知识：
$S[l][r]\times 10^{l-r}=s[l][k]-s[r][k]$
若$S[l][k]-S[r][k]\equiv 0(modP)$
$\because 10^{x}modP\ne 0$
$\therefore S[l][r]modP=0$
若$S[l][r]\equiv 0(modP)$
则$S[l][r]\times 10^{r-l}modP=0$

这道题再增加难度就是任意模数$P$了二不一定是$2019$，上面证明了任意模数P非10的因子的情况下。
下面讲如何处理$P$为$10$的因子情况下$:$
从$1$到$n$遍历若一个数个位为$10$的因子，则前面的数肯定是$10$的倍数，所以前面的数被$P$取余一定为$0$。
对应的ABC158E

将题意转化为对2019求余为0的子串有多少个。求一个后缀模数数组，后缀数组取模后的数为$i$,后缀数组取模后为$i$的个数为$M[i]$。
$i的贡献值=M[i]\ast (M[i]-1)/2$这里可以转化为加法，加上之前已有的$M[i]$之后再更新$M[i]$，注意初始化$M[0]=1$。

Code ABC164D

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int,int>P;
const double eps = 1e-8;
const int NINF = 0xc0c0c0c0;
const int INF  = 0x3f3f3f3f;
const ll  mod  = 2019;
const ll  maxn = 1e6 + 5;
const int N = 2e5 + 5;

char s[N];
int a[N],suf[N],n,M[2020];

int main(){
	ios::sync_with_stdio(false);
	cin.tie(0);
	cin>>s+1;
	int n=strlen(s+1);
	int cnt=0;int base=1;
	M[0]++;
	for(int i=n;i>=1;i--){
		a[i]=s[i]-'0';
		suf[i]=(suf[i+1]+a[i]*base)%mod;
		cnt+=M[suf[i]];
		M[suf[i]]++;
		base=base*10%mod;
	}
	cout<<cnt<<'\n';
	return 0;
}

Code2 ABC158E

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int,int>P;
const double eps = 1e-8;
const int NINF = 0xc0c0c0c0;
const int INF  = 0x3f3f3f3f;
const ll  mod  = 1e9 + 7;
const ll  maxn = 1e6 + 5;
const int N = 2e5 + 5;

char s[N];
int suf[N],cnt[N];

int main(){
	ios::sync_with_stdio(false);
	cin.tie(0);
	int n,p;cin>>n>>p>>s+1;
	int base=1;
	ll ans=0;
	if(10%p==0){
		for(int i=1;i<=n;i++){
			if((s[i]-'0')%p==0) ans+=i;
		}
	}else{
		cnt[0]++;
		for(int i=n;i>=1;i--){
			int j=s[i]-'0';
			suf[i]=(suf[i+1]+j*base)%p;
			ans+=cnt[suf[i]];
			cnt[suf[i]]++;
			base=base*10%p;
		}
	}
	cout<<ans<<'\n';
	return 0;
}