HDU 5145 NPY and girls 莫队+多重集排列

HDU 5145
题意:给出长度为n的序列a,m次询问:给出[L,R],为a[L],a[L+1]..a[R]组成的集合的排列方案是多少?
n,m,a[i]<=3e4.
每次询问的是多重集的排列方案?
回顾一下:[L,R]有k种不同数字 n1+n2+.nk=n 则假设排列方案为x 容易列出x*n1!*n2!..nk!=n!得x=n!/(n1!*..nk!).

容易知道增加和删除一个数以后答案的变化,离线莫队排序后处理一下.记录当前范围内的数的出现次数即可.

#include <bits/stdc++.h>using namespace std;typedef long long ll;const int N=3e4+50,mod=1e9+7;struct node{    int l,r,id;}q[N];int n,Q,a[N],pos[N],mk[N],l,r;ll res,f[N],inv[N],ans[N];ll powmod(ll x,ll n){    ll s=1;    while(n)    {        if(n&1)            s=(s*x)%mod;        x=(x*x)%mod;        n>>=1;    }    return s;}void init(){    l=1,r=0;    res=1;    f[0]=1,inv[0]=powmod(f[0],mod-2);    memset(mk,0,sizeof(mk));    for(ll i=1;i<N;i++)        f[i]=(f[i-1]*i)%mod,inv[i]=powmod(f[i],mod-2);    int block=sqrt(n);    for(int i=1;i<=n;i++)        pos[i]=(i-1)/block+1;}bool cmp(node a,node b){    if(pos[a.l]==pos[b.l])        return a.r<b.r;    return a.l<b.l;}void add(int p){    int m=r-l+1;    p=a[p];    res=(res*f[m])%mod;    res=(res*inv[m-1])%mod;    res=(res*f[mk[p]])%mod;    mk[p]++;    res=(res*inv[mk[p]])%mod;}void del(int p){    int m=r-l+1;    p=a[p];    res=(res*inv[m])%mod;    res=(res*f[m-1])%mod;    res=(res*f[mk[p]])%mod;    mk[p]--;    res=(res*inv[mk[p]])%mod;}void solve(){    for(int i=1;i<=Q;i++)    {        for(;r<q[i].r;)            add(++r);        for(;l>q[i].l;)            add(--l);        for(;r>q[i].r;r--)            del(r);        for(;l<q[i].l;l++)            del(l);        ans[q[i].id]=res;    }}int main(){    int T;    cin>>T;    while(T--)    {        scanf("%d%d",&n,&Q);        init();        for(int i=1;i<=n;i++)            scanf("%d",&a[i]);        for(int i=1;i<=Q;i++)            scanf("%d%d",&q[i].l,&q[i].r),q[i].id=i;        sort(q+1,q+1+Q,cmp);        solve();        for(int i=1;i<=Q;i++)            printf("%lld\n",ans[i]);    }    return 0;}