[莫队]hdu6959zoto

题目

题目
有一个序列，长度为$n$,序列为$（i,fx[i]）$，$m$次询问，询问以$(x1,y1)$作为左下坐标，$(x2,y2)$作为右上坐标形成的矩形内有几个序列内的点

思路

莫队，依照垂直$x$轴的最左端和左右端来分块，用树状数组来维护当前存在的点的数量

代码

#include<cstdio>
#include<cstring>
#include<cmath>
#include<cstdlib>
#include<cctype>
#include<ctime>
#include<iostream>
#include<string>
#include<map>
#include<queue>
#include<stack>
#include<set>
#include<vector>
#include<iomanip>
#include<list>
#include<bitset>
#include<sstream>
#include<fstream>
#include<complex>
#include<algorithm>
#if __cplusplus >= 201103L
#include <unordered_map>
#define lowbit(x) (x)&(-x)
#include <unordered_set>
#endif
#define ll long long
using namespace std;
const int INF = 0x3f3f3f3f;
const int N = 100010;
int fx[100010],b[N];
int vis[100010],ans[N],fen[N];

void modify(int x,int k) {
    for(int i=x; i<N; i+=lowbit(i)) b[i]+=k;
}

int get(int x) {
    int ans = 0;
    for(int i=x; i; i-=lowbit(i)){
    	ans+=b[i];
	}
    return ans;
}

int query(int x,int y) {
    return get(y)-get(x-1);
}

struct sut{
	int id,lx,ly,rx,ry; 
}a[100010];
bool cmp(sut m,sut n){
	return fen[m.lx]==fen[n.lx]?m.rx<n.rx:fen[m.lx]<fen[n.lx];
}
void add(int x){
	if(!vis[fx[x]]){
		modify(fx[x],1);
	}
	vis[fx[x]]++;
}
void delet(int x){
	if(vis[fx[x]]==1){
		modify(fx[x],-1);
	}
	vis[fx[x]]--;
}
int main(){
	ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
	int t;
	cin>>t;
	while(t--){
		memset(b,0,sizeof b);
		memset(vis,0,sizeof vis);
		int n,m;
		cin>>n>>m;
		for(int i=1;i<=n;i++){
			cin>>fx[i];
			fx[i]++;
		//	vis[fx[i]]=0;
			fen[i]=i/sqrt(n);
		}
		for(int i=1;i<=m;i++){
			int lx,ly,rx,ry;
			cin>>lx>>ly>>rx>>ry;
			ly++;
			ry++;//树状数组不能有0 
			a[i]={i,lx,ly,rx,ry};
		}
		sort(a+1,a+1+m,cmp);
		int xl=1,xr=0;
		for(int i=1;i<=m;i++){
		//	cout<<i<<" "<<xl<<" "<<xr<<" "<<a[i].lx<<" "<<a[i].ly<<" "<<a[i].rx<<" "<<a[i].ry<<" "<<a[i].id<<endl;
			while(a[i].lx<xl){
				add(--xl);
			}
			while(a[i].rx>xr){
				add(++xr);
			}
			while(a[i].lx>xl){
				delet(xl++);
			}
			while(a[i].rx<xr){
				delet(xr--);
			}
		//	cout<<i<<" "<<a[i].ly<<" "<<a[i].ry<<endl; 
			ans[a[i].id]=query(a[i].ly,a[i].ry);
		}
		for(int i=1;i<=m;i++){
			cout<<ans[i]<<endl;
		}
	}
    return 0;
}