P2709 小B的询问

合集 - 区间问题(37)

1.Iva & Pav2024-02-172.P2023 [AHOI2009] 维护序列2024-02-203.P1531 I Hate It2024-02-204.P5057 [CQOI2006] 简单题2024-02-205.P2357 守墓人2024-02-206.P8775 [蓝桥杯 2022 省 A] 青蛙过河2024-02-217.Dora and Search2024-02-248.P6492 [COCI2010-2011#6] STEP2024-02-269.D. Slimes2024-02-2810.P1040 [NOIP2003 提高组] 加分二叉树2024-03-0311.P2466 [SDOI2008] Sue 的小球2024-03-0412.P3957 [NOIP2017 普及组] 跳房子2024-03-0613.P2135 方块消除2024-03-0614.P1884 [USACO12FEB] Overplanting S2024-03-1015.P2642 双子序列最大和2024-03-2516.P2422 良好的感觉2024-03-2517.P3128 [USACO15DEC] Max Flow P2024-04-0218.P3258 [JLOI2014] 松鼠的新家2024-04-0219.P3384 【模板】重链剖分/树链剖分2024-04-0320.P5367 【模板】康托展开2024-04-0521.P2344 [USACO11FEB] Generic Cow Protests G2024-04-13

22.P2709 小B的询问2024-04-16

23.P4145 上帝造题的七分钟 2 / 花神游历各国2024-04-1724.P5677 [GZOI2017] 配对统计2024-04-1725.P2161 [SHOI2009] 会场预约2024-04-1726.F. Equal XOR Segments2024-05-0427.D. A BIT of an Inequality2024-05-0428.P3147 [USACO16OPEN] 262144 P2024-05-1329.P4290 [HAOI2008] 玩具取名2024-05-1330.E. Long Inversions2024-04-0931.P1668 [USACO04DEC] Cleaning Shifts S2024-05-1732.P9691 [GDCPC2023] Base Station Construction2024-05-1833.P2734 [USACO3.3] 游戏 A Game2024-06-0834.D. Sum of XOR Functions2024-06-1435.C. Medium Design2024-06-2836.F. Feed Cats2024-07-0737.E. Boring Segments2024-07-28

收起

原题链接

题解

莫队算法是局限性非常大的优化，离线+无修改，它通过邻近区间修改复杂度为 $O(1)$ 的特性让区间排序，然后再做修改，排序的规则是按块排序，然后左端点 $l$ 在一个块里的按右端点排序

code

#include<bits/stdc++.h>
using namespace std;

int a[50005];

struct node
{
    int l,r,id,kuai;
}q[50004];

int ans[50004];
int num[50004]={0};

bool cmp(node b,node c)
{
    if(b.kuai!=c.kuai) return b.kuai<c.kuai;
    else if(b.kuai&1) return b.r<c.r;
    else return b.r>c.r;
}

int main()
{
    int n,m,k;
    cin>>n>>m>>k;

    int sizes=sqrt(n);
    for(int i=1;i<=n;i++) cin>>a[i];

    for(int i=1;i<=m;i++)
    {
        cin>>q[i].l>>q[i].r;
        q[i].id=i;
        q[i].kuai=(q[i].l-1)/sizes+1;
    }

    sort(q+1,q+m+1,cmp);

    int sum=0;
    for(int i=q[1].l;i<=q[1].r;i++) num[a[i]]++;// 预处理第一个区间
    for(int i=1;i<=k;i++) sum+=num[i]*num[i];
    ans[q[1].id]=sum;

    int l=q[1].l,r=q[1].r;
    for(int i=2;i<=m;i++)
    {
        while(r<q[i].r){r++;sum+=2*num[a[r]]+1;num[a[r]]++;}//挪步
        while(l>q[i].l){l--;sum+=2*num[a[l]]+1;num[a[l]]++;}
        while(l<q[i].l){sum-=2*num[a[l]]-1;num[a[l]]--;l++;}
        while(r>q[i].r){sum-=2*num[a[r]]-1;num[a[r]]--;r--;}
        ans[q[i].id]=sum;
    }
    for(int i=1;i<=m;i++) cout<<ans[i]<<endl;
    return 0;
}