洛谷3245 [hnoi2016]大数

技不如人,莫队都不会写了

阅读全文大概需要 1分钟
本文总阅读量
Posted by yjjr's blog on February 27, 2018

标签:莫队

题目

题目传送门

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;
}
本文可以转载,但必须附上原文链接,否则你会终生找不到妹子!!!欢迎关注我的CSDN: ahyjjr