题目描述
小B有一个序列,包含N个1~K之间的整数。他一共有M个询问,每个询问给定一个区间[L..R],求Sigma(c(i)^2)的值,其中i的值从1到K,其中c(i)表示数字i在[L..R]中的重复次数。小B请你帮助他回答询问。
输入输出格式
输入格式:
第一行,三个整数N、M、K。
第二行,N个整数,表示小B的序列。
接下来的M行,每行两个整数L、R。
输出格式:
M行,每行一个整数,其中第i行的整数表示第i个询问的答案。
思路
莫队模板题
#include <bits/stdc++.h> using namespace std; const int maxn = 200000 + 10; int n,m,k,a[50000],ans[maxn],tot,cnt[1000001],block; struct Node { int l,r,num; inline bool operator < (Node cmp) const { if (l/block != cmp.l/block) return l/block < cmp.l/block; return r < cmp.r; } }p[maxn]; inline void add(int x) { cnt[a[x]]++; tot += 2*cnt[a[x]]-1; } inline void del(int x) { cnt[a[x]]--; tot -= 2*cnt[a[x]]+1; } int main() { scanf("%d%d%d",&n,&m,&k); block = sqrt(n); for (int i = 1;i <= n;i++) scanf("%d",&a[i]); for (int i = 1;i <= m;i++) scanf("%d%d",&p[i].l,&p[i].r),p[i].num = i; sort(p+1,p+m+1); int l = p[1].l,r = p[1].r; for (int i = l;i <= r;i++) add(i); ans[p[1].num] = tot; for (int i = 2;i <= m;i++) { while (l < p[i].l) del(l++); while (l > p[i].l) add(--l); while (r < p[i].r) add(++r); while (r > p[i].r) del(r--); ans[p[i].num] = tot; } for (int i = 1;i <= m;i++) printf("%d\n",ans[i]); return 0; }