Sona
给你一个序列,问你(L,R)区间内每个数出现次数的立方和(开始当成了平方和。。。)莫队模板题,需要离散化一下。
#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long ll;
const int maxn=100050;
int a[maxn],t,b[maxn];
ll c[maxn],an[maxn];
struct node{
int l,r,id;
}s[maxn];
bool cmp(const node &a,const node &b)
{
if(a.l/t==b.l/t) return a.r>b.r;
return a.l<b.l;
}
int n,m,N;
int main()
{
//freopen("in.txt","r",stdin);
while(~scanf("%d",&n))
{
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i]);
b[i]=a[i];
}
sort(b+1,b+n+1);
N=unique(b+1,b+n+1)-b-1;
for(int i=1;i<=n;i++)
{
a[i]=lower_bound(b+1,b+N+1,a[i])-b-1;
}
for(int i=0;i<=N+1;i++)
c[i]=0;
scanf("%d",&m);
for(int i=1;i<=m;i++)
{
scanf("%d%d",&s[i].l,&s[i].r);
s[i].id=i;
}
t=(int)(sqrt(1.0*n)+0.5);///这里写sqrt(n)会编译错误
sort(s+1,s+m+1,cmp);
int l=s[0].l,r=s[0].l-1;
ll ans=0;
//memset(c,0,sizeof c);
for(int i=1;i<=m;i++)
{
while(l<s[i].l)
{
ans-=c[a[l]]*c[a[l]]*c[a[l]];
c[a[l]]--;
ans+=c[a[l]]*c[a[l]]*c[a[l]];
l++;
}
while(l>s[i].l)
{
l--;
ans-=c[a[l]]*c[a[l]]*c[a[l]];
c[a[l]]++;
ans+=c[a[l]]*c[a[l]]*c[a[l]];
}
while(r<s[i].r)
{
r++;
ans-=c[a[r]]*c[a[r]]*c[a[r]];
c[a[r]]++;
ans+=c[a[r]]*c[a[r]]*c[a[r]];
}
while(r>s[i].r)
{
ans-=c[a[r]]*c[a[r]]*c[a[r]];
c[a[r]]--;
ans+=c[a[r]]*c[a[r]]*c[a[r]];
r--;
}
an[s[i].id]=ans;
}
for(int i=1;i<=m;i++)
printf("%I64d\n",an[i]);
}
return 0;
}