[BZOJ3236]:[Ahoi2013]作业(莫队+分块)
题目传送门
题目描述
此时已是凌晨两点,刚刚做了$Codeforces$的小$A$掏出了英语试卷。英语作业其实不算多,一个小时刚好可以做完。然后是一个小时可与做完的数学作业,接下来是分别都是一个小时可以做完的化学,物理,语文……小$A$压力巨大。
这时小$A$碰到了一道非常恶心的数学题,给定了一个长度为$n$的数列和若干个询问,每个询问是关于数列的区间$[l,r]$(表示数列的第$1$个数到第$r$个数),首先你要统计该区间内大于等于$a$,小于等于$b$的书的个数,其次是所有大于等于$a$,小于等于$b$的,且在该区间中出现过的数值的个数。
小$A$望着那数万的数据规模几乎绝望,只能向大神您求救,请您帮帮他吧。
输入格式
第一行两个数$n,m$,接下来$n$个数(这些数都大于等于$1$小于等于$n$),表示给定数列。
接下来$m$行,每行四个整数$l,r,a,b$:$l,r$表示询问的区间,$a,b$表示询问的数值的范围。
输出格式
输出$m$行,分别对应每个询问,输出两个数,分别为在l到r这段区间中大小在$[a,b]$中的数的个数,以及大于等于$a$,小于等于$b$的,且在该区间中出现过的数值的个数(具体可以参考样例)。
样例
样例输入
3 4
1 2 2
1 2 1 3
1 2 1 1
1 3 1 3
2 3 2 3
样例输出
2 2
1 1
3 2
2 1
数据范围与提示
$n=100,000$,$m=1,000,000$
题解
为了防止你们数错0,我专门在百万位和十万位之间加了“,”。
但是$n$却只有$100,000$,那么我们可以考虑莫队,至于统计答案,分块就好了。
不过正解好像是用权值线段树或者是树状数组,但是分块常数较小,跑得不比线段书慢。
代码时刻
#include<bits/stdc++.h>
using namespace std;
struct rec
{
int l;
int r;
int a;
int b;
int id;
int p;
}q[1000001];
int n,m;
int t,tt;
int a[1000001];
int lx[1000001],rx[1000001];
int cnt[1000001],pos[1000001],k[1000001],change[1000001];
int ans1[1000001],ans2[1000001];
bool cmp(rec a,rec b){return a.p==b.p?a.r<b.r:a.p<b.p;}//莫队
void get_answer(int l,int r,int id)//分块找答案
{
if(pos[l]==pos[r])
{
for(int i=l;i<=r;i++)
{
ans1[id]+=cnt[i];
if(cnt[i])ans2[id]++;
}
return;
}
if(pos[l])
for(int i=l;i<=rx[pos[l]];i++)
{
ans1[id]+=cnt[i];
if(cnt[i])ans2[id]++;
}
if(pos[r])
for(int i=lx[pos[r]];i<=r;i++)
{
ans1[id]+=cnt[i];
if(cnt[i])ans2[id]++;
}
if(pos[l]&&pos[r])
for(int i=pos[l]+1;i<pos[r];i++){ans1[id]+=k[i];ans2[id]+=change[i];}
if(pos[l]&&!pos[r])
for(int i=pos[l]+1;i<=tt;i++){ans1[id]+=k[i];ans2[id]+=change[i];}
}
int main()
{
scanf("%d%d",&n,&m);
t=sqrt(n);
tt=n/t;
if(n%t)tt++;
for(int i=1;i<=tt;i++)
{
lx[i]=(i-1)*t+1;
rx[i]=i*t;
}
rx[tt]=n;
for(int i=1;i<=n;i++)
scanf("%d",&a[i]);
for(int i=1;i<=m;i++)
{
scanf("%d%d%d%d",&q[i].l,&q[i].r,&q[i].a,&q[i].b);
q[i].id=i;
q[i].p=(q[i].l-1)/t+1;
pos[i]=(i-1)/t+1;
}
sort(q+1,q+m+1,cmp);
int l=1,r=0;
for(int i=1;i<=m;i++)
{
while(l<q[i].l)
{
cnt[a[l]]--;
k[pos[a[l]]]--;
if(!cnt[a[l]])change[pos[a[l]]]--;
l++;
}
while(l>q[i].l)
{
l--;
if(!cnt[a[l]])change[pos[a[l]]]++;
cnt[a[l]]++;
k[pos[a[l]]]++;
}
while(r<q[i].r)
{
r++;
if(!cnt[a[r]])change[pos[a[r]]]++;
cnt[a[r]]++;
k[pos[a[r]]]++;
}
while(r>q[i].r)
{
cnt[a[r]]--;
k[pos[a[r]]]--;
if(!cnt[a[r]])change[pos[a[r]]]--;
r--;
}
get_answer(q[i].a,q[i].b,q[i].id);
}
for(int i=1;i<=m;i++)
printf("%d %d\n",ans1[i],ans2[i]);
return 0;
}
rp++