题目
分析
一道典型的莫队。
先预处理出后缀,即f[i]表示i~(n-1) mod p 的值.
但p的值不小,显然不能直接存,加一个离散化。
观察题目,发现一串数s(l~r)整除p满足s(l~n-1)%p==s(r+1~n-1)%p
但p值为2或5不满足这个性质需要特判(不过数据中好像没有,于是笔者没写,有兴趣的可以自己去写写。。。。。。)
然后问题转化为求一段区间中有几对相等的f值。
套一个莫队。(ans(l,r)可以由【ans(l+1,r),ans(l-1,r),ans(l,r+1),ans(l,r-1)】推出,为保证时间复杂度,用个分块【因为不会曼哈顿树】)。
注意:要在数列的后面加一个f值为0。
代码
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cmath>
#include<cstring>
#define ll long long
using namespace std;
const int maxn=200000+10;
char a[maxn];
ll p,n,m,s[maxn],_10[maxn]/*_10[i]表示10的i次方modp*/,f[maxn]/*f[i]表示i~l mod p 的值*/,cd[maxn];
ll cur,sum[maxn],kuai,ans[maxn];
struct query
{
ll t,x,y;
}q[maxn];
//把每个区间(l,r)看成一个点(x,y),相同块中的点按y排序,否则按x排序,可以保证n^1.5的时间复杂度
bool operator < (const query& a,const query& b){
if(a.x/kuai==b.x/kuai) return a.y<b.y;
else return a.x<b.x;
}
int main()
{
//freopen("number.in","r",stdin);
//freopen("number.out","w",stdout);
cin>>p;
_10[0]=1;//其实这个_10数组可以只用一个变量
_10[1]=10%p;
for(int i=2;i<maxn;i++){
_10[i]=(_10[i-1]*_10[1])%p;
}
char c;
scanf("%s",a);
n=strlen(a);
kuai=(ll)sqrt((double)n);//按x坐标分块,kuai表示没段块的长度
cin>>m;
for(int i=1;i<=m;i++)
scanf("%lld%lld",&q[i].x,&q[i].y),q[i].x--,q[i].t=i;
sort(q+1,q+m+1);
for(int i=0;i<n;i++)
s[i]=a[i]-'0';
f[n-1]=s[n-1]%p;
cd[n-1]=f[n-1];
for(int i=n-2;i>=0;i--){
f[i]=(f[i+1]+(_10[n-i-1]*s[i])%p)%p;
cd[i]=f[i];
}
//for(int i=0;i<n;i++) printf("%lld ",f[i]);printf("\n");
//离散化
cd[n]=0;
sort(cd,cd+n+1);
int tot=0;
for(int i=1;i<=n;i++){
if(cd[i]!=cd[tot]) cd[++tot]=cd[i];
}
for(int i=0;i<n;i++)
{
int mid,l=0,r=tot;
while(l<r) //在ad中找到f的位置
{
mid=(l+r)>>1;
if(cd[mid]<f[i]) l=mid+1;
else r=mid;
}
f[i]=l;
}
//***********************************************
int l,r;
l=1;r=0;
for(int i=1;i<=m;i++)
{
while(r<q[i].y)
cur+=sum[f[++r]]++;
while(l>q[i].x)
cur+=sum[f[--l]]++;
while(r>q[i].y)
cur-=--sum[f[r--]];
while(l<q[i].x)
cur-=--sum[f[l++]];
ans[q[i].t]=cur;
}
for(int i=1;i<=m;i++) printf("%lld\n",ans[i]);
return 0;
}
/*
样例:
11
121121
3
1 6
1 5
1 4
*/