HNOI 2016 大数

大数

题意

1.有个长度为N的数S，可能有前导0
2.一个素数P
3.M个询问
4.S的一个子串中有多少子串是P的倍数(0也是P的倍数)
5.子串可以有前导0

解

是P的倍数，即Mod P=0

P30

定义
cnt[i][j]:[i,j]串中有cnt个子串
1.预处理cnt[i][j] O(n^2)
2.对于每个询问，枚举L，ans+=min(R[i],R)-L+1

Pac

1.特殊情况P=2||P=5 (10%P=0的情况)
只要序列末尾%P=0,这个序列就是P的倍数
证明：
设这个序列前面的数为A,最后一个数为B
这个序列表示成 $A*10+B$
由于 A*10%P=0 则 只要 B%P=0 即可

所以只要记录
① pre[i]:[1,i]有几个序列是P的倍数
② sum[i]:[1,i]有几个数%P=0
ans=pre[r]-pre[l-1]-(sum[r]-sum[l-1])*(l-1)

2.一般情况(10%P!=0的情况)
记A[i]=Pow(10,n-i)*(S[i]-‘0’)
记C[i]为A数组的后缀和
则一个序列[L,R]可以表示为
$(C[L]-C[R+1])/Pow(10,t)$
(t随L,R定，这里对结果没有影响所以没算)
目标是寻找
$(C[L]-C[R+1])/Pow(10,t)=0$的式子
化简即 $C[L]=C[R+1]$
问题转化成区间[L,R+1]中有多少个相同点对
因为每个L都要与R+1匹配，而R+1不匹配，所以询问区间转化为[L,R+1]

求区间相同点对，是莫队的经典问题

具体流程
当前已有[L,R]的答案，
加入一个x(L-1或R+1)，
对答案贡献是cnt[C[x]]，res+=cnt[C[x]]
然后cnt[C[x]]++
先res+，在cnt+
因为自己不能跟自己配对。
删除操作就是加的逆操作，先cnt-，在res-

这里C[x]比较大，但是个数只有n个，所以需要离散

具体代码

#include<bits/stdc++.h>
using namespace std;
const int M=100005;
int n,m,P,S;
char A[M];
int B[M],C[M];
long long ans[M],res,cnt[M];
struct Query {
    int id,l,r;
    bool operator <(const Query&tmp)const {
        if(l/S!=tmp.l/S)return l/S<tmp.l/S;
        return r<tmp.r;
    }
}Q[M];
void init() {
    int ten=1;
    for(int i=n-1;i>=0;i--){
        C[i]=1ll*ten*(A[i]-'0')%P;
        C[i]=(C[i]+C[i+1])%P;
        B[i]=C[i];
        ten=1ll*ten*10%P;
    }
    int len=n;
    B[len++]=0;
    sort(B,B+len);
    len=unique(B,B+len)-B;
    for(int i=0;i<n;i++){
        C[i]=lower_bound(B,B+len,C[i])-B;
    }
    C[n]=lower_bound(B,B+len,0)-B;
}
void add(int x){
    res+=cnt[C[x]]; 
    cnt[C[x]]++;
}
void del(int x){
    cnt[C[x]]--;
    res-=cnt[C[x]];
}
void mo() {
    int l=0,r=-1;
    for(int i=1; i<=m; i++) {
        while(r<Q[i].r)add(++r);
        while(r>Q[i].r)del(r--);
        while(l<Q[i].l)del(l++);
        while(l>Q[i].l)add(--l);
        ans[Q[i].id]=res;
    }
}
long long pre[M],sum[M];
void solve(){
    for(int i=n;i>=1;i--){
        A[i]=A[i-1];
    }
    for(int i=1;i<=n;i++){
        pre[i]=pre[i-1],sum[i]=sum[i-1];
        if((A[i]-'0')%P==0){
            pre[i]+=i;
            sum[i]++;
        }
    }
    int l,r;
    for(int i=1;i<=m;i++){
        scanf("%d %d",&l,&r);
        printf("%lld\n",pre[r]-pre[l-1]-(sum[r]-sum[l-1])*(l-1));
    }
}
int main() {
    scanf("%d %s %d",&P,A,&m);
    n=strlen(A);
    if(P==2||P==5){
        solve();
        return 0;
    }
    S=sqrt(n);
    init();
    for(int i=1; i<=m; i++) {
        scanf("%d %d",&Q[i].l,&Q[i].r);
        Q[i].l--;
        Q[i].id=i;
    }
    sort(Q+1,Q+1+m);
    mo();
    for(int i=1; i<=m; i++)printf("%lld\n",ans[i]);
    return 0;
}