二次离线莫队

    xiaoxiao2022-06-24  208

    例题:洛谷 P5047

    题面图上的妹子好可爱啊(ˉ﹃ˉ)

    多组询问,求一个区间内的逆序对数,离线, n ≤ 1 0 5 n \leq 10^5 n105,时限0.3s,空间限制32MB (好,不愧是lxl)

    二次离线莫队

    如果直接用莫队做这道题的话,每次移动都要来一遍树状数组,复杂度就是 O ( n n log ⁡ n ) O(n \sqrt{n} \log n) O(nn logn)的了。

    ( [ l 1 , r 1 ] , [ l 2 , r 2 ] ) ([l_1,r_1],[l_2,r_2]) ([l1,r1],[l2,r2])表示这两个区间之间的逆序对数。

    现在考虑做莫队的过程中,假如我要将区间 [ l , r ] [l,r] [l,r],通过右端点右移,变成区间 [ l , r ′ ] [l,r'] [l,r]。那么一次从 [ l , r + i ] [l,r+i] [l,r+i]变到 [ l , r + i + 1 ] [l,r+i+1] [l,r+i+1]的移动,当前答案就要加上 ( [ l , r + i ] , [ r + i + 1 , r + i + 1 ] ) ([l,r+i],[r+i+1,r+i+1]) ([l,r+i],[r+i+1,r+i+1])。差分一下,得到: ( [ 1 , r + i ] , [ r + i + 1 , r + i + 1 ] ) − ( [ 1 , l − 1 ] , [ r + i + 1 , r + i + 1 ] ) ([1,r+i],[r+i+1,r+i+1])-([1,l-1],[r+i+1,r+i+1]) ([1,r+i],[r+i+1,r+i+1])([1,l1],[r+i+1,r+i+1])

    因此这一次移动,当前答案应该加上:

    ( ∑ i = r + 1 r ′ ( [ 1 , i − 1 ] , [ i , i ] ) ) − ( [ 1 , l − 1 ] , [ r + 1 , r ′ ] ) (\sum_{i=r+1}^{r'} ([1,i-1],[i,i]))-([1,l-1],[r+1,r']) (i=r+1r([1,i1],[i,i]))([1,l1],[r+1,r])

    前者只有 n n n种情况,用树状数组 O ( n log ⁡ n ) O(n \log n) O(nlogn)预处理,然后利用前缀和计算。

    后者是一个前缀和一个区间的逆序对数的形式,将其离线存在每个前缀处。所有 [ r + 1 , r ′ ] [r+1,r'] [r+1,r]的总长,由于莫队的性质,是 O ( n n ) O(n \sqrt{n}) O(nn )级别的。那么可以从左到右处理每一个前缀以及挂在这个前缀上的 [ r + 1 , r ′ ] [r+1,r'] [r+1,r],利用分块,每次前缀加入一个新数,修改每个元素的值和前缀的逆序对数是 O ( n n ) O(n \sqrt{n}) O(nn )的,查询一个位置的元素与前缀的逆序对数是 O ( 1 ) O(1) O(1)的。

    总复杂度是 O ( n log ⁡ n + n n ) O(n \log n+n \sqrt{n}) O(nlogn+nn )的。

    其他的左端点左移,左端点右移,右端点左移也类似,左端点移动就用后缀即可。

    注意一点,莫队在处理完右端点的移动后,当前区间的右端点就变成了当前处理的询问的右端点,而不是上一个询问的右端点。

    代码

    #include<bits/stdc++.h> using namespace std; #define RI register int int read() { int q=0;char ch=' '; while(ch<'0'||ch>'9') ch=getchar(); while(ch>='0'&&ch<='9') q=q*10+ch-'0',ch=getchar(); return q; } typedef long long LL; const int N=100005,sqn=320; int n,m,js; int a[N],b[N],tr[N],pos[N],lz[sqn+5]; LL suml[N],sumr[N],ans[N],kansl[N],kansr[N]; struct node{int l,r,id;}q[N]; bool cmp(node A,node B) {return pos[A.l]==pos[B.l]?A.r<B.r:pos[A.l]<pos[B.l];} int lowbit(int x) {return x&(-x);} void add(int x) {while(x<=js) ++tr[x],x+=lowbit(x);} int query(int x) {int re=0;while(x) re+=tr[x],x-=lowbit(x); return re;} void prework() { for(RI i=1;i<=n;++i) suml[i]=suml[i-1]+query(js-a[i]),add(js-a[i]+1); for(RI i=1;i<=js;++i) tr[i]=0; for(RI i=n;i>=1;--i) sumr[i]=sumr[i+1]+query(a[i]-1),add(a[i]); } vector<node> lmove[N],rmove[N]; void work1() { sort(q+1,q+1+m,cmp); q[0].l=1,q[0].r=0; for(RI i=1;i<=m;++i) { if(q[i].r>q[i-1].r) lmove[q[i-1].l-1].push_back((node){q[i-1].r+1,q[i].r,i}); if(q[i].r<q[i-1].r) lmove[q[i-1].l-1].push_back((node){q[i].r+1,q[i-1].r,i}); if(q[i].l>q[i-1].l) rmove[q[i].r+1].push_back((node){q[i-1].l,q[i].l-1,i}); if(q[i].l<q[i-1].l) rmove[q[i].r+1].push_back((node){q[i].l,q[i-1].l-1,i}); } } void work2() { for(RI i=1;i<=js;++i) tr[i]=0; for(RI i=1;i<=pos[js];++i) lz[i]=0; for(RI i=0;i<n;++i) { int sz=lmove[i].size(); for(RI j=0;j<sz;++j) for(RI k=lmove[i][j].l;k<=lmove[i][j].r;++k) kansl[lmove[i][j].id]+=tr[a[k]]+lz[pos[a[k]]]; for(RI j=a[i+1]-1;j>=sqn*(pos[a[i+1]]-1)+1;--j) ++tr[j]; for(RI j=pos[a[i+1]]-1;j>=1;--j) ++lz[j]; } for(RI i=1;i<=js;++i) tr[i]=0; for(RI i=1;i<=pos[js];++i) lz[i]=0; for(RI i=n+1;i>=2;--i) { int sz=rmove[i].size(); for(RI j=0;j<sz;++j) for(RI k=rmove[i][j].l;k<=rmove[i][j].r;++k) kansr[rmove[i][j].id]+=tr[a[k]]+lz[pos[a[k]]]; for(RI j=a[i-1]+1;j<=sqn*pos[a[i-1]];++j) ++tr[j]; for(RI j=pos[a[i-1]]+1;j<=pos[js];++j) ++lz[j]; } } void work3() { LL nowans=0; for(RI i=1;i<=m;++i) { if(q[i].r>q[i-1].r) nowans+=suml[q[i].r]-suml[q[i-1].r]-kansl[i]; if(q[i].r<q[i-1].r) nowans-=suml[q[i-1].r]-suml[q[i].r]-kansl[i]; if(q[i].l>q[i-1].l) nowans-=sumr[q[i-1].l]-sumr[q[i].l]-kansr[i]; if(q[i].l<q[i-1].l) nowans+=sumr[q[i].l]-sumr[q[i-1].l]-kansr[i]; ans[q[i].id]=nowans; } } int main() { n=read(),m=read(); for(RI i=1;i<=n;++i) a[i]=b[i]=read(); sort(b+1,b+1+n); js=1;for(RI i=2;i<=n;++i) if(b[i]!=b[js]) b[++js]=b[i]; for(RI i=1;i<=n;++i) a[i]=lower_bound(b+1,b+1+js,a[i])-b; for(RI i=1;i<=m;++i) q[i].l=read(),q[i].r=read(),q[i].id=i; for(RI i=1;i<=n;++i) pos[i]=(i-1)/sqn+1; prework(),work1(),work2(),work3(); for(RI i=1;i<=m;++i) printf("%lld\n",ans[i]); return 0; }

    最新回复(0)