BZOJ 2038 小Z的袜子 莫队算法(模板题)

点击打开链接

莫队算法是离线处理一类区间不修改查询类问题的算法。就是如果你知道了[L,R]的答案。
你可以在O(1)的时间下得到[L,R-1]和[L,R+1]和[L-1,R]和[L+1,R]的答案的话。就可以使用莫队算法。

题意:n个数,m次询问,n,m<=5e4,每次询问:在区间[l,r]选两个数是相同概率? 
f[i]表示i在区间出现次数,答案为: S=segma(C(f[i],2)) /C(r-l+1,2) 
化解得 S=(segma(f[i]^2)-(r-l+1))/((r-l+1)*(r-l)
容易得到[L,R+1] S=S-f[z]^2+(f[z]+1)^2 ,f[z]++; 

参考链接

那么莫队算法怎么做呢？以下都是在转移为O(1)的基础下讨论的时间复杂度。另外由于n与m同阶，就统一写n。
如果已知[l,r]的答案，要求[l’,r’]的答案，我们很容易通过|l – l’|+|r – r’|次转移内求得。
将n个数分成sqrt(n)块。
按区间排序，以左端点所在块内为第一关键字，右端点为第二关键字，进行排序，也就是以(pos [l]，r)排序
然后按这个排序直接暴力，复杂度分析是这样的：
1、i与i+1在同一块内，r单调递增，所以r是O(n)的。由于有n^0.5块,所以这一部分时间复杂度是n^1.5。
2、i与i+1跨越一块，r最多变化n，由于有n^0.5块，所以这一部分时间复杂度是n^1.5
3、i与i+1在同一块内时l变化不超过n^0.5，跨越一块也不会超过n^0.5，忽略*2。由于有m次询问（和n同级），所以时间复杂度是n^1.5
于是就是O(n^1.5)了

#include <bits/stdc++.h>using namespace std;typedef long long ll;const int N=5e4+20; int n,m,c[N],pos[N];struct node{int l,r,id;ll a,b;//答案的分子和分母 }a[N];ll f[N],ans;ll gcd(ll a,ll b){if(a%b==0)return b;return gcd(b,a%b);}bool cmp(node a,node b){if(pos[a.l]==pos[b.l])return a.r<b.r;return a.l<b.l;}bool cmp_id(node a,node b){return a.id<b.id;}void init(){memset(f,0,sizeof(f));cin>>n>>m;for(int i=1;i<=n;i++)scanf("%d",&c[i]);int block=(int)sqrt(n);//分块for(int i=1;i<=n;i++)pos[i]=(i-1)/block+1;for(int i=1;i<=m;i++){scanf("%d%d",&a[i].l,&a[i].r);a[i].id=i;}}void update(int p,int add){ans-=f[c[p]]*f[c[p]];f[c[p]]+=add;//出现次数 ans+=f[c[p]]*f[c[p]];}void solve(){for(int i=1,l=1,r=0;i<=m;i++){for(;r<a[i].r;r++)update(r+1,1);for(;r>a[i].r;r--)update(r,-1);for(;l<a[i].l;l++)update(l,-1);for(;l>a[i].l;l--)update(l-1,1);if(a[i].l==a[i].r){a[i].a=0;a[i].b=1;continue;} a[i].a=ans-(a[i].r-a[i].l+1);//a[i].b=(ll)(a[i].r-a[i].l+1)*(a[i].r-a[i].l);//ll k=gcd(a[i].a,a[i].b);a[i].a/=k;a[i].b/=k;}}int main(){init();sort(a+1,a+1+m,cmp);solve();sort(a+1,a+1+m,cmp_id);for(int i=1;i<=m;i++)printf("%lld/%lld\n",a[i].a,a[i].b);return 0;}