题目链接:Problem - 6959
题意:给你一个长度为n的数列fx, fx[i]表示二维平面上有点(i,fx[i]),m次查询,每次给出一个矩形的左下坐标和右上坐标,求矩形内不同的y坐标个数。
分析:与普通莫队相比,多了矩形上下界的限制。考虑用莫队维护x坐标,用分块维护y坐标。
时间复杂度为O(n*sqrt(n)*1+m*1*sqrt(n)),莫队单次修改时O(sqrt(n))的复杂度遇上了值域上O(1)的修改,莫队单次查询时O(1)的复杂度遇上了值域查询时O(sqrt(n))的复杂度。
代码:
#include<bits/stdc++.h>
using namespace std;
const int maxn=1e5+5;
int a[maxn],num[maxn],sum[maxn],ans[maxn];
int k;
struct node{
int x0,y0,x1,y1,id;
}q[maxn];
bool cmp(node a,node b){//奇偶性排序
return (a.x0/k)^(b.x0/k)?a.x0<b.x0:(((a.x0/k)&1)?a.x1<b.x1:a.x1>b.x1);
}
void add(int x)
{
num[x]++;
if(num[x]==1) sum[x/k]++;
}
void dec(int x)
{
num[x]--;
if(num[x]==0) sum[x/k]--;
}
int calc(int x)
{
int now=0;
for(int i=0;i<x/k;i++) now+=sum[i];
for(int i=(x/k)*k;i<=x;i++)
if(num[i]) now+=1;
return now;
}
int main()
{
int t;
scanf("%d",&t);
while(t--)
{
memset(sum,0,sizeof num);
memset(num,0,sizeof sum);
int n,m;
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++) scanf("%d",&a[i]);
k=313;//对1e5开方
for(int i=1;i<=m;i++)
{
scanf("%d%d%d%d",&q[i].x0,&q[i].y0,&q[i].x1,&q[i].y1);
q[i].id=i;
}
sort(q+1,q+m+1,cmp);
int l=1,r=0;
for(int i=1;i<=m;i++)
{
int ql=q[i].x0,qr=q[i].x1;
while(l<ql) dec(a[l++]);
while(l>ql) add(a[--l]);
while(r<qr) add(a[++r]);
while(r>qr) dec(a[r--]);
ans[q[i].id]=calc(q[i].y1)-calc(q[i].y0-1);//前缀和操作
}
for(int i=1;i<=m;i++) printf("%d\n",ans[i]);
}
}