[BZOJ 2038] 2009国家集训队 小Z的袜子 · 莫队算法

今天看wzy大神的blog看见了莫队算法小结，心血来潮学了一下莫队算法的基础，没时间了啊还是打条模板题算了。 。。

以前没学过莫队的时候听机房里的小伙伴说什么有个算法复杂度是，特别不理解，原来还真有这种算法。。。跪莫涛大神

跪hzwer：我觉得这篇已经写得很好了   分块部分的思维值得研究

#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <algorithm>
#include <cmath>
using namespace std;
#define ll long long
#define sqr(x) ((x)*(x))

const int N=50005;
int n,m,cnt[N],a[N],tot;	//第i种颜色有cnt[i]个 
int l,r;
ll ans[N][2];
struct arr{
	int l,r,num;
}q[N];
int belong[N],num;
ll x,y,g,Ans;

int get(){
	int p=0;char x=getchar();
	while (x<'0' || x>'9') x=getchar();
	while (x>='0' && x<='9') p=p*10+x-'0',x=getchar();
	return p;
}

ll gcd(ll a,ll b){
	ll r;
	while (b) {r=a%b;a=b;b=r;}
	return a;
}

bool cmp(const arr A,const arr B){
	if (belong[A.l]==belong[B.l]) return A.r<B.r;
	return A.l<B.l;
}

void init(){
	
	int k=sqrt(m);
	
	tot=n/k;
	
	for (int i=1;i<=m;i++) belong[i]=(i-1)/k+1;	
	
}

void update(int p,int t){
	
	 Ans-=sqr(cnt[p]);
	 
	 cnt[p]+=t;
	 
	 Ans+=sqr(cnt[p]);
	
}

int main(){
	
	n=get();m=get();
	init();
	for (int i=1;i<=n;i++) a[i]=get();
	for (int i=1;i<=m;i++) q[i].l=get(),q[i].r=get(),q[i].num=i;
	
	sort(q+1,q+m+1,cmp);
	
	l=1; r=0; memset(cnt,0,sizeof cnt);
	
	for (int i=1;i<=m;i++){
		
		num=q[i].num;
		
		if (q[i].l==q[i].r) {
			ans[num][0]=0;ans[num][1]=1;
			continue;
		}
		
		for (;l<q[i].l;l++) update(a[l],-1);
		
		for (;l>q[i].l;l--)  update(a[l-1],1);
		
		for (;r<q[i].r;r++) update(a[r+1],1);
		
		for (;r>q[i].r;r--) update(a[r],-1);
		 
		x=Ans-(q[i].r-q[i].l+1);
		y=(ll)(q[i].r-q[i].l+1)*(q[i].r-q[i].l);
		g=gcd(x,y);
		ans[num][0]=x/g;
		ans[num][1]=y/g;
	
	}
	
	for (int i=1;i<=m;i++) 
		
		printf("%lld/%lld\n",ans[i][0],ans[i][1]);
		
	return 0;
	
}