题意:
求区间[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;
}
京公网安备 11010502036488号