1. 程式人生 > >【題解】Luogu SP3267 DQUERY - D-query

【題解】Luogu SP3267 DQUERY - D-query

原題傳送門

這題和Luogu P1972 [SDOI2009]HH的項鍊很像,只是資料大小有些差別,題解

我部落格裡對莫隊的介紹

我們在排序詢問時,普通是這樣qaq

inline bool cmp(register query a,register query b)
{
    return a.bl==b.bl?a.r<b.r:a.bl<b.bl;
}

我加了一個奇偶優化:按奇偶塊排序。如果區間左端點所在塊不同,那麼就直接按左端點從小到大排;如果相同,奇塊按右端點從小到大排,偶塊按右端點從大到小排。

inline bool cmp(register query a,register query b)
{
    return a.bl!=b.bl?a.l<b.l:((a.bl&1)?a.r<b.r:a.r>b.r);
}

剩下我們就按照莫隊的板子模擬就行啦qaq

完整程式碼:

#pragma GCC optimize("O3")
#include <bits/stdc++.h>
#define N 500005
using namespace std;
inline int read()
{
    register int x=0,f=1;register char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9')x=(x<<3)+(x<<1)+ch-'0',ch=getchar();
    return x*f;
}
inline void write(register int x)
{
    if(!x)putchar('0');if(x<0)x=-x,putchar('-');
    static int sta[25];int tot=0;
    while(x)sta[tot++]=x%10,x/=10;
    while(tot)putchar(sta[--tot]+48);
}
struct query{
    int l,r,id,bl;
}q[N];
int num[1000001],blocksize=0,a[N],ans[N];
inline bool cmp(register query a,register query b)
{
    return a.bl!=b.bl?a.l<b.l:((a.bl&1)?a.r<b.r:a.r>b.r);
}
int main()
{
    int n=read();
    blocksize=sqrt(n);
    for(register int i=1;i<=n;++i)
        a[i]=read();
    int m=read();
    for(register int i=1;i<=m;++i)
    {
        int l=read(),r=read();
        q[i]=(query){l,r,i,l/blocksize};
    }
    sort(q+1,q+1+m,cmp);
    int l=1,r=0,sum=0;
    for(register int i=1;i<=m;++i)
    {
        int ll=q[i].l,rr=q[i].r;
        while(ll<l)
            sum+=!num[a[--l]]++;
        while(ll>l)
            sum-=!--num[a[l++]];
        while(rr>r)
            sum+=!num[a[++r]]++;
        while(rr<r)
            sum-=!--num[a[r--]];
        ans[q[i].id]=sum;
    }
    for(register int i=1;i<=m;++i)
        write(ans[i]),printf("\n");
    return 0;
}