题目:点击此处
题意:求一个区间内出现的数的次数的立方的和
思路:主要不同就是add和remove里的不同
add:当前某个数的出现次数是cnt,如果遇见了这个数,那么先让ans-=cnt*cnt*cnt,然后ans+=(cnt+1)*(cnt+1)*(cnt+1);
remove:当前某个数的出现次数是cnt,如果遇见了这个数,那么先让ans-=cnt*cnt*cnt,然后ans+=(cnt-1)*(cnt-1)*(cnt-1);
由于给的n个数字比较大,n比较小,所以必须离散化
注意不能用map直接操作,这样会超时
#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <queue>
#include <vector>
#include <list>
#include <map>
#include <set>
#define ll long long
using namespace std;
const int maxn=1e5+5;
int n,q,a[maxn],block,l,r,b[maxn];
ll cnt[maxn],answer[maxn],ans;
map<int,int> mp;
struct node
{
int l,r,i;
}pos[maxn];
bool cmp(node a,node b)
{
if(a.l/block!=b.l/block)
return a.l/block<b.l/block;
return a.r<b.r;
}
ll cube(ll x)
{
return x*x*x;
}
void add(int x)
{
ans-=cube(cnt[a[x]]);
cnt[a[x]]++;
ans+=cube(cnt[a[x]]);
}
void remove1(int x)
{
ans-=cube(cnt[a[x]]);
cnt[a[x]]--;
ans+=cube(cnt[a[x]]);
}
int main()
{
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);
memset(cnt,0,sizeof(cnt));
int m=unique(b+1,b+1+n)-b-1;
ans=0;
for(int i=1;i<=n;i++)
{
a[i]=lower_bound(b+1,b+1+n,a[i])-b-1;
}
block=sqrt(1.0*m);
scanf("%d",&q);
for(int i=1;i<=q;i++)
{
scanf("%d%d",&pos[i].l,&pos[i].r);
pos[i].i=i;
}
sort(pos+1,pos+1+q,cmp);
ll cl=pos[1].l,cr=cl-1;
for(int i=1;i<=q;i++)
{
l=pos[i].l,r=pos[i].r;
while(cl<l)
{
remove1(cl++);
}
while(cl>l)
{
add(--cl);
}
while(cr<r)
{
add(++cr);
}
while(cr>r)
{
remove1(cr--);
}
answer[pos[i].i]=ans;
}
for(int i=1;i<=q;i++)
printf("%lld\n",answer[i]);
}
return 0;
}