NBUT-1457 Sona

Sona

莫队 模板题

多组输入

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <vector>
#include <string>
#include <queue>
#include <functional>
#include <map>
#include <set>
#include <cmath>
#include <cstring>
#include <deque>
#include <stack>
using namespace std;
typedef long long ll;
#define pii pair<int, int>
const ll maxn = 1e5 + 10;
const ll inf = 1e17 + 10;
int num[maxn], cnt[maxn], pos[maxn], alp[maxn], a[maxn];
ll ans = 0, last[maxn];

struct node
{
    int l, r, id;
    node(){}
    node(int _l, int _r, int _id){l = _l; r = _r; id = _id;}
}seg[maxn];

bool cmp(const node& a, const node& b)
{
    return pos[a.l] ^ pos[b.l] ? pos[a.l] < pos[b.l] : (pos[a.l] & 1 ? a.r > b.r : a.r < b.r);
}

inline ll f(ll x)
{
    return x * x * x;
}

inline void add(int x)
{
    ans -= f(cnt[num[x]]);
    ans += f(++cnt[num[x]]);
}

inline void del(int x)
{
    ans -= f(cnt[num[x]]);
    ans += f(--cnt[num[x]]);
}

int main()
{
    int n,m;
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    while(cin >> n)
    {
        ans = 0;
        for(int i=1; i<=n; i++) {cin >> num[i]; a[i] = num[i];}
        int tp = 0;
        sort(a + 1, a + n + 1);
        a[0] = a[1] - 1;
        for(int i=1; i<=n; i++) if(a[i] != a[i-1]) alp[tp++] = a[i];
        for(int i=1; i<=n; i++) num[i] = lower_bound(alp, alp + tp, num[i]) - alp;
        int t = (double)sqrt((double)n);
        for(int i=1; i<=n; i++) pos[i] = i / t;

        cin >> m;
        for(int i=0; i<m; i++)
        {
            int x, y;
            cin >> x >> y;
            seg[i] = node(x, y, i);
        }
        sort(seg, seg + m, cmp);
        int l = 1, r = 0;
        for(int i=0; i<m; i++)
        {
            while(r < seg[i].r) add(++r);
            while(l > seg[i].l) add(--l);
            while(r > seg[i].r) del(r--);
            while(l < seg[i].l) del(l++);
            last[seg[i].id] = ans;
        }
        for(int i=0; i<m; i++)
        {
            cout << last[i] << endl;
        }
        for(int i=0; i<=tp; i++) cnt[i] = 0;
    }
    return 0;
}