NBUT 1457 Sona (莫队算法)

题意：

给你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]);
    }

}