bzoj 5016 [Snoi2017]一个简单的询问

题面

https://www.lydsy.com/JudgeOnline/problem.php?id=5016

题解

莫队算法

因此我们可以用(l,r) 表示ask(1,l,1,r) 然后就可以了莫队了

用num1,num2两个数组记录下当前两个区间中每个数字分别出现多少次

Code

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;

ll read(){
    ll x=0,f=1;char c=getchar();
    while(c<'0' || c>'9'){if(c=='-')f=-1;c=getchar();}
    while(c>='0' && c<='9'){x=x*10+c-'0';c=getchar();}
    return x*f;
}

const int maxn=50050;
const int sz=250;
int n;
int a[maxn];
int Q,tot;
struct query{
    int l,r,id,ad;
    bool operator < (query ano) const{
        if(l/sz==ano.l/sz) return r<ano.r;
        else return l<ano.l;
    }
} q[maxn*10];

inline void add(int l,int r,int id,int ad){
    q[++tot].l=l,q[tot].r=r;
    q[tot].id=id,q[tot].ad=ad;
}
int num1[maxn],num2[maxn],ans[maxn];

int main(){
#ifdef LZT
    freopen("in","r",stdin);
#endif
    n=read();
    for(int i=1;i<=n;i++) a[i]=read();
    Q=read();
    for(int i=1;i<=Q;i++){
        int l1,r1,l2,r2;
        l1=read(),r1=read(),l2=read(),r2=read();
        add(r1,r2,i,1);
        add(l1-1,r2,i,-1);
        add(r1,l2-1,i,-1);
        add(l1-1,l2-1,i,1);
    }
    sort(q+1,q+tot+1);
    int L=0,R=0,res=0;
    for(int i=1;i<=tot;i++){
        while(L<q[i].l){L++;res+=num2[a[L]];num1[a[L]]++;}
        while(L>q[i].l){res-=num2[a[L]];num1[a[L]]--;L--;}
        while(R<q[i].r){R++;res+=num1[a[R]];num2[a[R]]++;}
        while(R>q[i].r){res-=num1[a[R]];num2[a[R]]--;R--;}
        ans[q[i].id]+=res*q[i].ad;
    }
    for(int i=1;i<=Q;i++)
        printf("%d
",ans[i]);
    return 0;
}

/*
5
1 1 1 1 1
2
1 2 3 4
1 1 4 4
*/

Review

这样做的动机?

(数据范围50000=>应该用$O(n sqrt n)$的做法=>莫队)

(可离线的区间问题=>莫队)

应该不算难想到