洛谷P2709 BZOJ 3781 小B的询问 （莫队）

题目描述

小B有一个序列，包含N个1~K之间的整数。他一共有M个询问，每个询问给定一个区间[L..R]，求Sigma(c(i)^2)的值,其中i的值从1到K，其中c(i)表示数字i在[L..R]中的重复次数。小B请你帮助他回答询问。

输入输出格式

输入格式：

 

第一行，三个整数N、M、K。

第二行，N个整数，表示小B的序列。

接下来的M行，每行两个整数L、R。

 

输出格式：

 

M行，每行一个整数，其中第i行的整数表示第i个询问的答案。

 

输入输出样例

输入样例#1： 复制 
6 4 3
1 3 2 1 1 3
1 4
2 6
3 5
5 6
输出样例#1： 复制 
6
9
5
2

说明

对于全部的数据，1<=N、M、K<=50000

题解

　　第二次了……打莫队没分块……T飞了……

　　据说这题模拟可过……

　　考虑莫队，只要把每一次更改前的次数减掉，再加上更改后的次数就行了

 1 //minamoto
 2 #include<iostream>
 3 #include<cstdio>
 4 #include<algorithm>
 5 #include<cmath>
 6 #define ll long long
 7 using namespace std;
 8 #define getc() (p1==p2&&(p2=(p1=buf)+fread(buf,1,1<<21,stdin),p1==p2)?EOF:*p1++)
 9 char buf[1<<21],*p1=buf,*p2=buf;
10 inline int read(){
11     #define num ch-'0'
12     char ch;bool flag=0;int res;
13     while(!isdigit(ch=getc()))
14     (ch=='-')&&(flag=true);
15     for(res=num;isdigit(ch=getc());res=res*10+num);
16     (flag)&&(res=-res);
17     #undef num
18     return res;
19 }
20 char sr[1<<21],z[20];int C=-1,Z;
21 inline void Ot(){fwrite(sr,1,C+1,stdout),C=-1;}
22 inline void print(ll x){
23     if(C>1<<20)Ot();if(x<0)sr[++C]=45,x=-x;
24     while(z[++Z]=x%10+48,x/=10);
25     while(sr[++C]=z[Z],--Z);sr[++C]='\n';
26 }
27 const int N=50005;
28 int cnt[N],a[N],rt[N],l,r,n,m,k,s;ll ans[N],ansn;
29 struct node{
30     int l,r,id;
31 }q[N];
32 inline bool cmp(node a,node b){
33     return rt[a.l]==rt[b.l]?rt[a.l]&1?a.r<b.r:a.r>b.r:a.l<b.l;
34 }
35 inline void add(int x){
36     ansn-=1ll*cnt[x]*cnt[x];
37     ++cnt[x];
38     ansn+=1ll*cnt[x]*cnt[x];
39 }
40 inline void del(int x){
41     ansn-=1ll*cnt[x]*cnt[x];
42     --cnt[x];
43     ansn+=1ll*cnt[x]*cnt[x];
44 }
45 int main(){
46     //freopen("testdata.in","r",stdin);
47     n=read(),m=read(),k=read(),s=sqrt(n);
48     for(int i=1;i<=n;++i) a[i]=read(),rt[i]=(i-1)/s+1;
49     for(int i=1;i<=m;++i)
50     q[i].l=read(),q[i].r=read(),q[i].id=i;
51     sort(q+1,q+1+m,cmp);
52     l=1,r=0;
53     for(int i=1;i<=m;++i){
54         while(l>q[i].l) add(a[--l]);
55         while(r<q[i].r) add(a[++r]);
56         while(l<q[i].l) del(a[l++]);
57         while(r>q[i].r) del(a[r--]);
58         ans[q[i].id]=ansn; 
59     }
60     for(int i=1;i<=m;++i) print(ans[i]);
61     Ot();
62     return 0;
63 }

 

posted @ 2018-08-26 12:48  bztMinamoto  阅读(177)  评论(0)  编辑  收藏  举报
Live2D