分块模板(莫队算法)

时间复杂度;n*sqrt(n)

#include <map>
#include <set>
#include <stack>
#include <queue>
#include <cmath>
#include <ctime>
#include <vector>
#include <cstdio>
#include <cctype>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <algorithm>
using namespace std;
#define INF 0x3f3f3f3f
#define inf -0x3f3f3f3f
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define mem0(a) memset(a,0,sizeof(a))
#define mem1(a) memset(a,-1,sizeof(a))
#define mem(a, b) memset(a, b, sizeof(a))
typedef long long ll;
const int maxn=100000+10;
int a[maxn];
struct query{
    int l,r,id,pos;
    bool operator <(const query &t) const{
        return pos!=t.pos ? pos<t.pos:r<t.r;
    }
};
query b[maxn];
int ans[maxn];
int res;
int vis[maxn];

void reduce(int tmp){
    if(vis[tmp-1]+vis[tmp+1]==2){
        res++;
    }
    if(vis[tmp-1]+vis[tmp+1]==0){
        res--;
    }
    vis[tmp]=0;
}

void add(int tmp){
    if(vis[tmp-1]+vis[tmp+1]==2){
        res--;
    }
    if(vis[tmp-1]+vis[tmp+1]==0){
        res++;
    }
    vis[tmp]=1;
}

int main(){
    int t,n,m;
    scanf("%d",&t);
    while(t--){
        scanf("%d%d",&n,&m);
        for(int i=1;i<=n;i++)
            scanf("%d",&a[i]);
        int unit=(int) sqrt(n*1.0);
        for(int i=1;i<=m;i++){
            scanf("%d%d",&b[i].l,&b[i].r);
            b[i].pos=b[i].l/unit;
            b[i].id=i;
        }
        sort(b+1,b+m+1);
        mem0(vis);
        int tmp;
        int l=1,r=0;
        res=0;
        for(int i=1;i<=m;i++){
            while(r>b[i].r){
                tmp=a[r];
                reduce(tmp);
                r--;
            }
            while(r<b[i].r){
                r++;
                tmp=a[r];
                add(tmp);
            }
            while(l>b[i].l){
                l--;
                tmp=a[l];
                add(tmp);
            }
            while(l<b[i].l){
                tmp=a[l];
                reduce(tmp);
                l++;
            }
            ans[b[i].id]=res;
        }
        for(int i=1;i<=m;i++){
            printf("%d\n",ans[i]);
        }
    }
    return 0;
}

你可能感兴趣的:(分块模板(莫队算法))