【bzoj1485:】【 [HNOI2009]有趣的数列】模任意数的卡特兰数

【bzoj1485:】【 [HNOI2009]有趣的数列】模任意数的卡特兰数_第1张图片
(上不了p站我要死了,侵权度娘背锅)

Description
我们称一个长度为2n的数列是有趣的,当且仅当该数列满足以下三个条件:
(1)它是从1到2n共2n个整数的一个排列{ai};
(2)所有的奇数项满足a1 < a3 < … < a2n-1,所有的偶数项满足a2 < a4 < … < a2n;
(3)任意相邻的两项a2i-1与a2i(1≤i≤n)满足奇数项小于偶数项,即:a2i-1 < a2i。
现在的任务是:对于给定的n,请求出有多少个不同的长度为2n的有趣的数列。因为最后的答案可能很大,所以只要求输出答案 mod P的值。
Input
输入文件只包含用空格隔开的两个整数n和P。输入数据保证,50%的数据满足n≤1000,100%的数据满足n≤1000000且P≤1000000000。
Output
仅含一个整数,表示不同的长度为2n的有趣的数列个数mod P的值。
Sample Input
3 10
Sample Output
5
对应的5个有趣的数列分别为(1,2,3,4,5,6),(1,2,3,5,4,6),(1,3,2,4,5,6),(1,3,2,5,4,6),(1,4,2,5,3,6)。

首先这是一个卡特兰数

我们思考把1~n的数字按顺序摆放。因为分了奇数和偶数,就相当于分了两组队列,我们把这n个数按顺序放入两组队列里。由于任何时候奇数队列里的元素个数都大于等于偶数队列里的,所以是卡特兰的一个经典模型。

答案即为C(2n,n)/(n+1) mod p

但是发现p并不是一个质数,所以不能求逆元,就不能直接用阶乘的公式来做。

这里就有一个很重要的方法了
虽然是模任意数,但是n的范围是o(n)可过的。我们筛出需要的数(如n)的阶乘(n!)中每个素数的指数。具体做法为:
设cnt[i]表示该阶乘(n!)中包含多少个i,进行线性递推。初始值cnt[i]=1。
依次枚举n到1,跳过质数。设minp[i]为i的最小质因子,cnt[i/minp[i]]+=cnt[i], cnt[minp[i]]+=cnt[i], cnt[i]=0。
这部分的代码(init()为线性筛):

init();
for(int i=n*2;i>=2;i--){
    cnt[i]++;
    if(!notp[i]) continue;
    cnt[i/minp[i]]+=cnt[i];
    cnt[minp[i]]+=cnt[i];
    cnt[i]=0;
}

这样就将除法问题转化为了质数的相减问题,得出C(2n,n)/(n+1)中每个质数的指数,再做快速幂。
时间复杂度:根据某素数定理,n中的素数个数约为 n/ln n 个,快速幂是 log n 的。所以乘起来约为o(n)

完整代码:

#include
#include
#include
using namespace std;
#define ll long long 
#ifdef WIN32
#define RIN "%I64d"
#else 
#define RIN "%lld"
#endif

const int N=1000000+5;

int cnt1[N*2],cnt2[N*2],cnt3[N*2];
int notp[N*2],cntp=0,prime[N*2],minp[N*2];
int n,p;

void init(){
    notp[1]=1;
    for(int i=2;i<=n*2;i++){
        if(!notp[i])
            prime[++cntp]=i;
        for(int j=1;j<=cntp&&i*prime[j]<=n*2;j++){
            notp[i*prime[j]]=1;
            minp[i*prime[j]]=prime[j];
            if(i%prime[j]==0) break;
        }
    }
}
ll power(int a,int b){
    ll rt=1;
    for(;b;b>>=1,a=(a*a)%p)
        if(b&1) rt=(rt*a)%p;
    return rt;
}
int main(){
    scanf("%d%d",&n,&p);
    init();
    for(int i=n*2;i>=2;i--){
        cnt1[i]++;
        if(!notp[i]) continue;
        cnt1[i/minp[i]]+=cnt1[i];
        cnt1[minp[i]]+=cnt1[i];
        cnt1[i]=0;
    }
    for(int i=n;i>=2;i--){
        cnt2[i]++;
        if(!notp[i]) continue;
        cnt2[i/minp[i]]+=cnt2[i];
        cnt2[minp[i]]+=cnt2[i];
        cnt2[i]=0;
    }
    for(int i=n+1;i>=2;i--){
        cnt3[i]++;
        if(!notp[i]) continue;
        cnt3[i/minp[i]]+=cnt3[i];
        cnt3[minp[i]]+=cnt3[i];
        cnt3[i]=0;
    }
    ll ans=1;
    for(int i=1;i<=cntp;i++){
        int tmp=prime[i];//printf("%d %d %d\n",cnt1[tmp],cnt2[tmp],cnt3[tmp]);
        ans=(ans*power(prime[i],cnt1[tmp]-cnt2[tmp]-cnt3[tmp]))%p;
    }
    printf(RIN"\n",ans);
    return 0;
}

你可能感兴趣的:(题目总结,数论)