莫队算法

oiwiki

img

普通莫队算法

P2709 小B的询问

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
#include<bits/stdc++.h>
using namespace std;
using u32 = unsigned;
#define i128 __int128;
using ll = long long;
//#define int ll
using u64 = unsigned long long;
const ll inf = 1e9;
const ll INF = 1e18;
const int N=5e4+10;
int bl;
ll sum;
struct node
{
    int l,r,id;
}v[N];
int a[N],cnt[N];
bool cmp(node &a,node&b)
{
    if(a.l/bl!=b.l/bl)return a.l<b.l;
    return a.r<b.r;
}
void add(int x)
{
    sum+=2*(cnt[x]++)+1;
}
void del(int x)
{
    sum+=-2*(cnt[x]--)+1;
}
signed main()
{
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    int n,m,k;
    cin>>n>>m>>k;
    bl=sqrt(n);
    for(int i=1;i<=n;i++)
    {
        cin>>a[i];
    } 
    for(int i=1;i<=m;i++)   
    {
        cin>>v[i].l>>v[i].r;
        v[i].id=i;
    }   
    vector<int>ans(m+1);
    sort(v+1,v+1+m,cmp);
    for(int i=1,l=1,r=0;i<=m;i++)
    {
        while(l>v[i].l)add(a[--l]);
        while(r<v[i].r)add(a[++r]);
        while(l<v[i].l)del(a[l++]);
        while(r>v[i].r)del(a[r--]);
        ans[v[i].id]=sum;
    }
    for(int i=1;i<=m;i++)cout<<ans[i]<<'\n';

    return 0;    
}