题意:
求区间[l,r]内每个数出现次数的平方*这个数,然后求和是多少
num(i)为i出现的次数
题解:
暴力固然不可行
其实这个题不难看出有莫队的感觉,之前做过统计数字出现次数的题,就是用莫队,把那个add和del改一改就行
不过这个题有点恶心,卡常
我们一般块的大小时用sqrt(n)来决定,这个要开到600以上才能过
代码好打,但是不好过,要尽可能优化下不然容易被卡
代码:
#include<cstdio> #include<iostream> #include<fstream> #include<algorithm> #include<functional> #include<cstring> #include<string> #include<cstdlib> #include<iomanip> #include<numeric> #include<cctype> #include<cmath> #include<ctime> #include<queue> #include<stack> #include<list> #include<set> #include<map> using namespace std; const int maxn=2000010; typedef long long ll; int n,m,k; struct node { int l,r,id; }Q[maxn]; int pos[maxn]; bool cmp(const node &a,const node &b) { if(pos[a.l]==pos[b.l]) return a.r<b.r; else return pos[a.l]<pos[b.l]; } ll num[maxn*2]; ll a[maxn],b[maxn],c[maxn]; ll ans[maxn],ans2[maxn]; ll Ans=0; void add(int x) { Ans-=num[a[x]]*num[a[x]]*a[x]; num[a[x]]++; Ans+=num[a[x]]*num[a[x]]*a[x]; } void del(int x) { Ans-=num[a[x]]*num[a[x]]*a[x]; num[a[x]]--; Ans+=num[a[x]]*num[a[x]]*a[x]; } int main() { while(~scanf("%d%d",&n,&m)) { memset(num,0,sizeof(num)); int block=600; for(int i=1;i<=n;i++) { scanf("%lld",&a[i]); pos[i]=i/block; } for(int i=1;i<=m;i++) { scanf("%d%d",&Q[i].l,&Q[i].r); Q[i].id=i; } sort(Q+1,Q+1+m,cmp); num[0]=0; int L=0,R=0; Ans=0; for(int i=1;i<=m;i++) { while(R<Q[i].r) { R++; add(R); } while(L>Q[i].l) { L--; add(L); } while(L<Q[i].l) { del(L); L++; } while(R>Q[i].r) { del(R); R--; } ans[Q[i].id]=Ans; } for(int i=1;i<=m;i++) printf("%I64d\n",ans[i]); } return 0; }