NBUT 1457 Sona (莫队算法)

来源:互联网 发布:对json进行url编码 编辑:程序博客网 时间:2024/05/18 19:44

题意:

给你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]);    }}
原创粉丝点击