P1972 [SDOI2009]HH的项链(莫队)

题目大意:

给你长度为n的一组数,每次询问区间有多少个不同的数。
N<100000 M<100000。

思路:

听说暴力能过,但是有点不科学,于是我打了一个莫队。莫队思想就是把左区间分块,然后暴力,这样就可以优化到Osqrt(O).

程序:

#include<cstdio>
#include<iostream>
#include<cmath>
#include<cstring>
#include<algorithm>
#define N 100000
using namespace std;
struct data{int l,r,qv,id;}q[N];
int n,m,sum=0;
int a[N],ans[N],num[N];
int cmp(const data&a,const data&b){
    if (a.qv<b.qv) return 1;
    if (a.qv>b.qv) return 0;
    if (a.r<b.r) return 1;
    return 0;
}

int add(int x){
    num[x]++;
    if (num[x]==1) sum++;
}

int del(int x){
    num[x]--;
    if (!num[x]) sum--;
}

int main(){
    scanf("%d",&n); int limit=(int)(sqrt((double)(n)+0.5));
    for (int i=1;i<=n;i++)
        scanf("%d",&a[i]);
    scanf("%d",&m);
    for (int i=1;i<=m;i++){
        scanf("%d%d",&q[i].l,&q[i].r);
        q[i].qv=q[i].l/limit; q[i].id=i;    
    }   
    sort(q+1,q+m+1,cmp);
    int l=0,r=0;
    for (int i=1;i<=m;i++){
        while (r<q[i].r){
            r++;
            add(a[r]);
        }
        while (r>q[i].r){
            del(a[r]);
            r--;
        }
        while (l>q[i].l){
            l--;
            add(a[l]);
        }
        while (l<q[i].l){
            del(a[l]);
            l++;
        }
        ans[q[i].id]=sum;
    }
    for (int i=1;i<=m;i++)
        printf("%d\n",ans[i]);
}

猜你喜欢

转载自blog.csdn.net/qq872425710/article/details/79449999