莫队 模板题
多组输入
#include #include #include #include #include #include #include #include #include #include #include #include #include using namespace std; typedef long long ll; #define pii pair 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> x >> y; seg[i] = node(x, y, i); } sort(seg, seg + m, cmp); int l = 1, r = 0; for(int i=0; i 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