洛谷3245 [HNOI2016]大数

标签：莫队

题目

题目传送门

Description

　　小 B 有一个很大的数 S，长度达到了 N 位；这个数可以看成是一个串，它可能有前导 0，例如00009312345。小B还有一个素数P。现在，小 B 提出了 M 个询问，每个询问求 S 的一个子串中有多少子串是 P 的倍数（0 也是P 的倍数）。例如 S为0077时，其子串 007有6个子串：0,0,7,00,07,007；显然0077的子串007有6个子串都是素数7的倍数。

Input

　　第一行一个整数：P。第二行一个串：S。第三行一个整数：M。接下来M行，每行两个整数 fr,to，表示对S 的子串S[fr…to]的一次询问。注意：S的最左端的数字的位置序号为 1；例如S为213567，则S[1]为 2，S[1…3]为 213。N,M<=100000，P为素数

Output

　　输出M行，每行一个整数，第 i行是第 i个询问的答案。
　　

Sample Input

11 
121121 
3 
1 6 
1 5 
1 4 

Sample Output

5
3
2

HINT

//第一个询问问的是整个串，满足条件的子串分别有：121121,2112,11,121,121。

2016.4.19新加数据一组

题意

给定一个由数字组成的字符串和模数P，M次询问，每次询问给定两个端点L，R

询问L-R中，有多少个子串，满足其代表的数字能够被模数P整除

分析

先推式子（好烦qwq）

$$ans=\sum_{i=l}^r \sum_{j=i}^r [(\sum_{k=i}^j st[k]*10^{j-k})\mod p ==0]$$

$$ans=\sum_{i=l}^r \sum_{j=i}^r [(10^j*\sum_{k=i}^j st[k]*10^{-k})\mod p ==0]$$

我们设

$$num[i]=\sum_{j=i}^nst[j]*(10')^j$$

$$ans=\sum_{i=l}^r\sum_{j=i}^r[10^j*(num[i]-num[j+1])\mod p==0]$$

可以从后往前推一遍得到num和sum数组

然后就转化为询问l->r中，有多少对num[i]==num[j]

可以用离散化+离线算法莫队解决

因为题目保证了p为质数，所以当p=2或p=5的时候需要特判

用两个数组分别表示1->i中2或5的倍数时有多少种情况及有多少个数末尾是2或5的倍数，用前缀和维护即可

code

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<map>
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define dep(i,a,b) for(int i=a;i>=b;i--)
#define ll long long
#define mem(x,num) memset(x,num,sizeof x)
#define reg(x) for(int i=last[x];i;i=e[i].next)
using namespace std;
inline ll read()
{
    ll f=1,x=0;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
const int maxn=4e5+6;
ll sqn,num[maxn],sum[maxn],ba[maxn],ans[maxn];
ll n,m,p,l=1,r=0,s=0,Bit=1;
char st[maxn];map<ll,ll>Map;
struct node{ll l,r,id;}ask[maxn];
inline bool operator < (node a,node b){return a.l/sqn==b.l/sqn ? a.r<b.r : a.l/sqn<b.l/sqn;}
int main()
{
    p=read();scanf("%s",st+1);m=read();
    n=strlen(st+1);sqn=(ll)sqrt(n*1.0);
    if(p!=2&&p!=5){
        dep(i,n,1){
            Bit=Bit*10%p;//每位累乘 
            num[i]=(num[i+1]+(st[i]-48)*Bit)%p; 
            sum[i]=num[i];//从末尾开始的字符串后缀hash转化为数字 
        }
        sort(sum+1,sum+1+n);//后缀排序 
        rep(i,1,n+1)Map[sum[i]]=i;
        rep(i,1,n+1)num[i]=Map[num[i]];
        rep(i,1,m)ask[i].l=read(),ask[i].r=read()+1,ask[i].id=i;
        sort(ask+1,ask+1+m);//对询问进行排序 
        rep(i,1,m){
            while(r<ask[i].r)s+=ba[num[++r]]++;
            while(l>ask[i].l)s+=ba[num[--l]]++;
            while(l<ask[i].l)s-=--ba[num[l++]];
            while(r>ask[i].r)s-=--ba[num[r--]];
            ans[ask[i].id]=s;//莫队并统计答案 
        }
        rep(i,1,m)printf("%lld\n",ans[i]);
    }else{
        rep(i,1,n)
            if(!((st[i]-48)%p))ba[i]=ba[i-1]+1,sum[i]=sum[i-1]+i;else ba[i]=ba[i-1],sum[i]=sum[i-1];//直接用前缀和维护2和5的特殊情况 
        rep(i,1,m)l=read(),r=read(),printf("%lld\n",sum[r]-sum[l-1]-(ba[r]-ba[l-1])*(l-1));
    }
    return 0;
}