#莫队,分块#洛谷 4396 bzoj 3236 作业

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/sugar_free_mint/article/details/82154115

题目

给定了一个长度为n的数列和若干个询问,每个询问是关于数列的区间表示数列的第1个数到第r个数),首先你要统计该区间内大于等于a,小于等于b的数的个数,其次是所有大于等于a,小于等于b的,且在该区间中出现过的数值的个数。


分析

根据这道题的复杂的询问,相对于线段树和树状数组是很难解决的,那么就需要用分块维护,然而用了分块还不够,于是莫队就上场了,当然对于神奇的优化暴力就不多说了


代码

#include <cstdio>
#include <cmath>
#include <algorithm>
struct rec{int l,r,a,b,rk;}g[100001]; int b[100001],cnt[100001],t;
int a[100001],ans1[100001],ans2[100001],f[100001][2],n,m,l=1,r;
int in(){
    int ans=0; char c=getchar();
    while (c<48||c>57) c=getchar();
    while (c>47&&c<58) ans=ans*10+c-48,c=getchar();
    return ans;
}
bool cmp(const rec &x,const rec &y){return b[x.l]==b[y.l]?x.r<y.r:b[x.l]<b[y.l];}//分块思想
void add(int x){
    f[b[x]][0]++; cnt[x]++;//大块和小块都要维护
    if (cnt[x]==1) f[b[x]][1]++;
}
void del(int x){
    f[b[x]][0]--; cnt[x]--;
    if (cnt[x]==0) f[b[x]][1]--;
}
void answ(int l,int r,int x){
    if (b[l]==b[r]){//小段朴素
        for (register int i=l;i<=r;i++)
            if (cnt[i]) ans1[x]+=cnt[i],ans2[x]++;
    }
    else{
        for (register int i=b[l]*t-1;i>=l;i--) if (cnt[i]) ans1[x]+=cnt[i],ans2[x]++;
        for (register int i=b[r]*t-t;i<=r;i++) if (cnt[i]) ans1[x]+=cnt[i],ans2[x]++;
        for (register int i=b[l]+1;i<b[r];i++) ans1[x]+=f[i][0],ans2[x]+=f[i][1];//大段维护
    }
}
int main(){
    n=in(); m=in(); t=sqrt(n);
    for (register int i=1;i<=n;i++) a[i]=in(),b[i]=i/t+1;
    for (register int i=1;i<=m;i++) g[i]=(rec){in(),in(),in(),in(),i};
    std::sort(g+1,g+1+m,cmp);
    for (register int i=1;i<=m;i++){
        while (l>g[i].l) add(a[--l]);//莫队模板
        while (l<g[i].l) del(a[l++]);
        while (r<g[i].r) add(a[++r]);
        while (r>g[i].r) del(a[r--]);
        answ(g[i].a,g[i].b,g[i].rk);//求答案
    }
    for (register int i=1;i<=m;i++) printf("%d %d\n",ans1[i],ans2[i]);
    return 0;
}

猜你喜欢

转载自blog.csdn.net/sugar_free_mint/article/details/82154115
今日推荐