[BZOJ1485] [HNOI2009]有趣的数列 && 找规律 + 数学

通过打表找规律可以发现这玩意是个卡特兰数列  那么问题转化为求卡特兰数列第n项h(n)

然后百度告诉我这玩意等于C(n*2, n) / (n+1) 那么问题就在于mod了 mod并不一定是一个质数 逆元并不一定存在 所以我们选择在计算过程中将组合数进行质因数分解 

那么问题就在于 如何快速的进行组合数分解  我们可以在O(n)的时间内递推出每一个数的一个因子 那么我们在计算时记录每一个因子的次数 如果该因子为质数 则直接乘入答案

如果该因子不为质数 我们就领cnt[Div[i]] += cnt[i], cnt[i/Div[i]] += cnt[i]; 这样就分解成了两个因子 直到分解为全为质因子的情况

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<iostream>
#include<queue>
#define SF scanf
#define PF printf
using namespace std;
typedef long long LL;
const int MAXN = 2000000;
int Div[MAXN+10], prime[MAXN+10], tot, MOD, n;
int cnt[MAXN+10];
bool vis[MAXN+10];
void init() {
    for(int i = 2; i <= MAXN; i++) {
        if(!vis[i]) prime[++tot] = i;
        for(int j = 1; i * prime[j] <= MAXN; j++) {
            Div[i*prime[j]] = prime[j];
            vis[i*prime[j]] = true;
            if(i % prime[j] == 0) break;
        }
    }
}
// h(n) = C(2n, n) / (n+1)
int pow_mod(int x, int k) {
    LL ans = 1;
    while(k) {
        if(k & 1) ans = 1LL * ans * x % MOD;
        x = 1LL * x * x % MOD;
        k >>= 1;
    }
    return ans;
}
int main() {
    init();
    SF("%d%d", &n, &MOD);
    for(int i = 2; i <= n; i++) cnt[i] = -1;
    for(int i = n+2; i <= 2*n; i++) cnt[i] = 1;
    int ans = 1;
    for(int i = n*2; i >= 2; i--)
        if(!vis[i]) ans = 1LL * ans * pow_mod(i, cnt[i]) % MOD;
        else cnt[Div[i]] += cnt[i], cnt[i/Div[i]] += cnt[i];
    PF("%d", ans);
}