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=0;
    while(Cget>'9'||Cget<'0')Cget=getchar();
    while(Cget>='0'&&Cget<='9')
    {
        now=now*10+Cget-'0';
        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=1;i<=n;i++) in(ai[i]),bel[i]=(i+1)/blo;
    for(int i=1;i<=m;i++) in(qu[i].l),in(qu[i].r),qu[i].id=i;
    sort(qu+1,qu+m+1);int l=1,r=0;
    for(int no=1;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=1;i<=m;i++) printf("%lld
",ans[i]);
    return 0;
}