题目描述
输入
输出
有M行,每个询问一行,输出结果mod 1,000,000,007的值。
样例输入
10 3
3 5 1 2 3 1 3 5 2 1
7 9
3 9
2 3
样例输出
10
19
6
数据范围
对于30%的数据,N,M<=1000
对于50%的数据,N,M<=30000
对于100%的数据,N,M<=100000
解法
离线不修改区间询问,考虑莫队算法。
利用线性筛法预处理出所有要用的逆元后。
显然每次容易O(1)处理。
总的时间复杂度为O(n1.5)。
代码
#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#define ll long long
#define sqr(x) ((x)*(x))
#define ln(x,y) ll(log(x)/log(y))
using namespace std;
const char* fin="ex3568.in";
const char* fout="ex3568.out";
const ll inf=0x7fffffff;
const ll maxn=100007,mo=1000000007;
ll n,m,i,j,k,ks,l,r,tmp;
ll a[maxn],ans[maxn],tong[maxn],tx[maxn];
ll ny[maxn];
struct qu{
ll l,r,lk,id;
}b[maxn];
bool cmp(qu a,qu b){
return a.lk<b.lk || a.lk==b.lk && a.r<b.r;
}
ll qpower(ll a,ll b){
ll c=1;
while (b){
if (b&1) c=c*a%mo;
a=a*a%mo;
b>>=1;
}
return c;
}
void add(ll v){
tmp=(tmp+mo-tx[a[v]])%mo;
if (tong[a[v]]==0) tx[a[v]]=1;
tong[a[v]]++;
tx[a[v]]=(tx[a[v]]*a[v])%mo;
tmp=(tmp+tx[a[v]])%mo;
}
void del(ll v){
tmp=(tmp+mo-tx[a[v]])%mo;
tong[a[v]]--;
tx[a[v]]=(tx[a[v]]*ny[a[v]])%mo;
if (tong[a[v]]==0) tx[a[v]]=0;
tmp=(tmp+tx[a[v]])%mo;
}
int main(){
scanf("%d%d",&n,&m);
ks=(ll)(sqrt(n));
for (i=1;i<=n;i++) scanf("%d",&a[i]);
for (i=1;i<=100000;i++) ny[i]=qpower(i,mo-2);
for (i=1;i<=m;i++){
scanf("%d%d",&b[i].l,&b[i].r);
b[i].lk=(b[i].l-1)/ks+1;
b[i].id=i;
}
sort(b+1,b+m+1,cmp);
l=1;
r=0;
tmp=0;
for (i=1;i<=m;i++){
while (r<b[i].r) add(++r);
while (l>b[i].l) add(--l);
while (l<b[i].l) del(l++);
while (r>b[i].r) del(r--);
ans[b[i].id]=tmp;
}
for (i=1;i<=m;i++) printf("%lld\n",ans[i]);
return 0;
}
启发
涉及到预处理素数取模时的逆元,可由这道题对线筛的分析得到预处理逆元的可能性。
莫队的单次扩展一定要保证O(1)的时间。