题意:
给你n个数字,q个询问,问[l,r]的价值,价值是每个数字出现次数的立方和
思路:
很基础的莫队,可是,这个oj有点毒
首先对于[l,r],我们转移为[l,r+1]的时候,只要加上去
cnt[id[arr[r]]]3
即可,其他的转移也很类似,也很好想,就不多说了。但是要注意这里需要离散化,并且不能直接用map,vector直接调用,要用id数组记录下id,否则会T
错误及反思:
其实上面的都是废话,因为这就是个莫队模板题,但是!这个OJ是真的毒,首先不能用bits/stdc++.h,而且sqrt()必须在math.h里面才可以,不能是cmath,还有就是要用I64d输出,不能用lld,而且返回的错误有时候有点偏差。。
代码:
2781ms:
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<math.h>
#include<vector>
using namespace std;
const int N = 501000;
int id[N],pos[N],n,m,arr[N];
long long ans[N];
long long cnt[N];
long long now=0;
vector<int> v;
struct Q
{
int l,r,id;
}q[N];
int getid(int x){return lower_bound(v.begin(),v.end(),x)-v.begin();}
bool cmp(Q a,Q b)
{
if(pos[a.l]!=pos[b.l])
return a.l<b.l;
return a.r<b.r;
}
inline void gx(int y,int x)
{
now-=1ll*cnt[y]*cnt[y]*cnt[y];
cnt[y]+=x;
now+=1ll*cnt[y]*cnt[y]*cnt[y];
}
int main()
{
while(scanf("%d",&n)!=EOF)
{
memset(cnt,0,sizeof(cnt));
now=0;
int block=sqrt(n);
for(int i=1;i<=n;i++)
{
scanf("%d",&arr[i]);
pos[i]=i/block;
v.push_back(arr[i]);
}
sort(v.begin(),v.end()),v.erase(unique(v.begin(),v.end()),v.end());
for(int i=1;i<=n;i++)
id[i]=getid(arr[i]);
scanf("%d",&m);
for(int i=0;i<m;i++)
{
scanf("%d%d",&q[i].l,&q[i].r);
q[i].id=i;
}
sort(q,q+m,cmp);
int l=1,r=0;
for(int i=0;i<m;i++)
{
for(;l<q[i].l;l++) gx(id[l],-1);
for(;l>q[i].l;l--) gx(id[l-1],1);
for(;r<q[i].r;r++) gx(id[r+1],1);
for(;r>q[i].r;r--) gx(id[r],-1);
ans[q[i].id]=now;
}
for(int i=0;i<m;i++)
printf("%I64d\n",ans[i]);
}
}