HDU-6959-zoto(莫队+分块）

HDU-6959

题意：给一个f数组，f[i]表示横坐标为i的纵坐标，给出m次询问，每次给出一个(x0,y0),(x1,y1)，问这两个点围成的矩形中有多少个不同的y坐标
思路：我们对横坐标使用莫队，对纵坐标使用分块，用num[i]表示纵坐标为i的个数，用sum[i]表示第i个块内有几个不同的纵坐标，对每个询问用前缀和思想求A® - A(l-1)即可，时间复杂度n*sqrt(n)

#include<bits/stdc++.h>
using namespace std;
#define inf 0x3f3f3f3f
#define IO ios::sync_with_stdio(false)
#define bug cout << "-----\n"
typedef long long ll;
int Mod = 998244353;
const int N = 100100;
const int M = 500010;
int a[N],num[N],sum[N],arr[N],block,xb;
struct node {
    int x1,y1,x2,y2,id;
    bool operator < (const node a) const {
        if(x1/xb!=a.x1/xb)return x1 < a.x1;
        return x2 < a.x2;
    }
}p[N];
void add(int x) {
    int fx = a[x];
    num[fx] ++;
    if(num[fx]==1)sum[fx/block] ++;
}
void del(int x) {
    int fx = a[x];
    num[fx] --;
    if(num[fx]==0)sum[fx/block] --;
}
int solve(int x) {
    int i,res = 0;
    for(i=0;i<x/block;i++) {
        res += sum[i];
    }
    for(i=(x/block)*block;i<=x;i++)
        if(num[i])res ++;
    return res;
}
int main() {
    IO;
    int T,i,j,n,m;
    cin >> T;
    while(T--) {
        memset(num,0,sizeof num);
        memset(sum,0,sizeof sum);
        int Max = 0;
        cin >> n >> m;
        for(i=1;i<=n;i++) {
            cin >> a[i];
            if(a[i] > Max)Max = a[i];
        }
        block = sqrt(Max);xb = sqrt(n);
        for(i=1;i<=m;i++) {
            cin >> p[i].x1 >> p[i].y1 >> p[i].x2 >> p[i].y2;
            p[i].id = i;
        }
        sort(p+1,p+1+m);
        int l = 1,r = 0;
        for(i=1;i<=m;i++) {
            int fl = p[i].x1,fr = p[i].x2;
            while(l<fl)del(l++);
            while(l>fl)add(--l);
            while(r<fr)add(++r);
            while(r>fr)del(r--);
            arr[p[i].id] = solve(p[i].y2) - solve(p[i].y1-1);
        }
        for(i=1;i<=m;i++) {
            cout << arr[i] << endl;
        }
    }
    return 0;
}