AC日记——小B的询问 洛谷 P2709

小B的询问

思路：

　　水题；

代码：

#include <bits/stdc++.h>

using namespace std;

#define maxn 50005

#define ll long long

int bel[maxn];

struct QueryType {

    int l,r,id;

    bool operator <(const QueryType pos)const

    {

        if(bel[l]==bel[pos.l]) return r<pos.r;

        else return bel[l]<bel[pos.l];

    }

};

struct QueryType qu[maxn];

int n,m,k,num[maxn],ai[maxn],blo;

ll now,ans[maxn];

inline void in(int &now)

{

    char Cget=getchar();now=;

    while(Cget>''||Cget<'')Cget=getchar();

    while(Cget>=''&&Cget<='')

    {

        now=now*+Cget-'';

        Cget=getchar();

    }

}

inline void updata(int x,bool di)

{

    x=ai[x];

    now-=num[x]*num[x];

    if(di) num[x]++,now+=num[x]*num[x];

    else num[x]--,now+=num[x]*num[x];

}

int main()

{

    in(n),in(m),in(k),blo=sqrt(n);

    for(int i=;i<=n;i++) in(ai[i]),bel[i]=(i+)/blo;

    for(int i=;i<=m;i++) in(qu[i].l),in(qu[i].r),qu[i].id=i;

    sort(qu+,qu+m+);int l=,r=;

    for(int no=;no<=m;no++)

    {

        while(r<qu[no].r) updata(++r,true);

        while(r>qu[no].r) updata(r--,false);

        while(l<qu[no].l) updata(l++,false);

        while(l>qu[no].l) updata(--l,true);

        ans[qu[no].id]=now;

    }

    for(int i=;i<=m;i++) printf("%lld\n",ans[i]);

    return ;

}