bzoj1485: [HNOI2009]有趣的数列

链接：http://www.lydsy.com/JudgeOnline/problem.php?id=1485

题意：中文题。。。

分析：题目需要我们将1~2*n这些数填入长度为2*n的数组并且要求(1)奇数位是递增，(2)偶数为也是递增，(3)相邻的a[2*i-1]<a[2*i]。咋一看要求好像有点多？首先我们想办法处理掉前两个条件：我们将这2*n个数从小到大填入数组，如果当前我们要填入的是奇数位那么我们填入第一个没有被填数的奇数位，偶数位同理。那么第三个条件怎么办呢？我们只要在填数的时候再加一个条件即可："当填到i时，在奇数位上的数的个数要大于等于在偶数位上的数"，这样我们就可以保证第三个条件的每一对数都是先填奇数位再填偶数位啦！这样有什么用呢？这不就是卡特兰数么。算一下C(2*n,n)/(n+1)即可。但是这里还有另外一个问题，模数p不一定为质数，那么组合数的除法就不能用逆元了，怎么办呢？那么只能用分解质因数来处理掉1~2*n了，那么这样的复杂度是O(nlogn)，对于n=10^6还是能过的。但是还有一种O(n)的方法，还记得O(n)的素数筛法么，有人在这上面做了点记录就能使1~2*n的质因数分解变成O(n)，因为O(n)筛法的原理我们可以知道每一个i的最小的质因数是谁，那么我们把他记录下来为b[i]，那么我们在做*i的时候就能变成*b[i]*(i/b[i])了，这样一直处理下去就能变成O(n)啦。

代码：

#include<map>
#include<set>
#include<cmath>
#include<queue>
#include<math.h>
#include<cstdio>
#include<vector>
#include<string>
#include<cstring>
#include<iostream>
#include<algorithm>
#pragma comment(linker, "/STACK:102400000,102400000")
using namespace std;
const int N=2000010;
const int MAX=151;
const int MOD1=100000007;
const int MOD2=100000009;
const double EPS=0.00000001;
typedef long long ll;
const ll MOD=1000000007;
const ll INF=10000000010;
typedef unsigned long long ull;
int a[N],b[N],q[N],num[N];
void deal(int n) {
    int i,j,k=0;
    memset(q,0,sizeof(q));
    for (i=2;i<=n;i++) {
        if (!q[i]) { a[++k]=i;b[i]=i; }
        for (j=1;j<=k;j++) {
            if (a[j]*i>n) break ;
            q[a[j]*i]=1;b[a[j]*i]=a[j];
            if (i%a[j]==0) break ;
        }
    }
}
ll qpow(ll a,ll b,ll p) {
    ll ret=1;
    while (b) {
        if (b&1) ret=ret*a%p;
        a=a*a%p;
        b>>=1;
    }
    return ret;
}
int main()
{
    int i,n;
    ll p,ans=1;
    scanf("%d%lld", &n, &p);
    deal(2*n);
    memset(num,0,sizeof(num));
    for (i=1;i<=n;i++) num[i]--;
    for (i=n+2;i<=2*n;i++) num[i]++;
    for (i=2*n;i>1;i--)
    if (!q[i]) continue ;
    else {
        num[b[i]]+=num[i];num[i/b[i]]+=num[i];num[i]=0;
    }
    for (i=2*n;i>1;i--) ans=ans*qpow(i,num[i],p)%p;
    printf("%lld\n", (ans+p)%p);
    return 0;
}