hdu_5145_NPY and girls(莫队算法+组合)

题目连接：http://acm.hdu.edu.cn/showproblem.php?pid=5145

题意：给你n,m，共有n个女孩，标号为1—n，n个数xi表示第ith个女孩在第xi个教室，然后下面有m个询问，每个询问有l,r两个数，表示要去找编号为l到r的女孩，每进一次教室只能找一个女孩，问有多少种组合方式，不同的组合方式定义为去教室的顺序不同。

题解：这题是离线的，很容易想到用莫队分块做，主要是要找出状态转移的方程，对于询问区间[l,r]，假设这个区间的女孩一共分布在k个教室，每个教室有a1,a2,a3....ak个女孩

则这个区间的ans=C(r-l+1,a1)*C(r-l+1-a1,a2)......C(r-l+1-a1-a2..-ak-1,ak)，整理方程,消去约数 得ans=(r-l+1)!/a1!*a2!*a3!...ak!。然后这时候 当r=r+1时，假设r+1这个女孩在a1这个教室，那么此时的ans用上面的式子算出来得ans=(r-l+2)!/(a1+1)*a1!*a2!*a3!...ak!。我们可以发现 ans([l,r+1])=ans([l,r])*(r-l+2)/(a1+1)。然后状态转移的方程就出来了，最后再预处理一下逆元，就可以开始出答案了。

#include<cstdio>#include<algorithm>#include<cmath>#define F(i,a,b) for(int i=a;i<=b;i++)using namespace std;const int N=3e4+7,mod=1e9+7;//逆元筛long long inv[N]={0,1};void init(){F(i,2,N-1)inv[i]=inv[mod%i]*(mod-mod/i)%mod;}int t,n,m,a[N],sqr,cnt[N],ans[N];//sqr为分块的大小，cnt记录的对应教室的女孩数//以左边边界所在的块为首关键字，左边边界所在的块为第二关键字排序struct qy{int l,r,lid,idx;bool operator<(const qy& b)const{if(lid==b.lid)return r<b.r;else return lid<b.lid;}}q[N];int main(){init(),scanf("%d",&t);while(t--){scanf("%d%d",&n,&m),sqr=(int)sqrt(1.0*n);F(i,1,n)scanf("%d",a+i);F(i,0,N-1)cnt[i]=0;F(i,1,m){scanf("%d%d",&q[i].l,&q[i].r);q[i].lid=q[i].l/sqr,q[i].idx=i;}sort(q+1,q+1+m);int l=1,r=0,len=0;long long ret=1;F(i,1,m){while(r<q[i].r){r++,len++,cnt[a[r]]++;ret=(ret*len%mod)*inv[cnt[a[r]]]%mod;}while(r>q[i].r){ret=(ret*cnt[a[r]]%mod)*inv[len]%mod;cnt[a[r]]--,r--,len--;}while(l>q[i].l){l--,len++,cnt[a[l]]++;ret=(ret*len%mod)*inv[cnt[a[l]]]%mod;}while(l<q[i].l){ret=(ret*cnt[a[l]]%mod)*inv[len]%mod;cnt[a[l]]--,l++,len--;}ans[q[i].idx]=ret;}F(i,1,m)printf("%d\n",ans[i]);}return 0;}