[BZOJ3907]网格（卡特兰数+组合数学+高精度）

题目描述

传送门

题解

这道题和买票找零问题是差不多一样的
计算的时候需要高精度
分解质因数之后只需要一个高精乘单精和一个高精减就可以了

代码

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
using namespace std;
#define N 10000

int n,m;
int prime[N+1],p[N+1],cnt[N+1],num[N+1];
struct data{int a[N+1];}ans1,ans2,ans;

void get()
{
    for (int i=2;i<=N;++i)
    {
        if (!p[i]) prime[++prime[0]]=i,num[i]=prime[0];
        for (int j=1;j<=prime[0]&&i*prime[j]<=N;++j)
        {
            p[i*prime[j]]=1;
            if (i%prime[j]==0) break;
        }
    }
}
data cheng(data a,int b)
{
    data ans;memset(ans.a,0,sizeof(ans.a));
    int len=a.a[0];
    for (int i=1;i<=len;++i) ans.a[i]=a.a[i]*b;
    for (int i=1;i<=len;++i)
    {
        ans.a[i+1]+=ans.a[i]/10;
        ans.a[i]%=10;
    }
    while (ans.a[len+1])
    {
        ++len;
        ans.a[len+1]+=ans.a[len]/10;
        ans.a[len]%=10;
    }
    ans.a[0]=len;
    return ans;
}
data jian(data a,data b)
{
    data ans;memset(ans.a,0,sizeof(ans.a));
    int len=a.a[0];
    for (int i=1;i<=len;++i)
    {
        if (a.a[i]<b.a[i]) --a.a[i+1],a.a[i]+=10;
        ans.a[i]=a.a[i]-b.a[i];
    }
    while (!ans.a[len]) --len;
    ans.a[0]=len;
    return ans;
}

int main()
{
    get();
    scanf("%d%d",&n,&m);
    for (int i=n+1;i<=n+m;++i)
    {
        int x=i;
        for (int j=1;j<=prime[0]&&x>1&&prime[j]*prime[j]<=x;++j)
            while (x%prime[j]==0) ++cnt[j],x/=prime[j];
        if (x>1) ++cnt[num[x]];
    }
    for (int i=1;i<=m;++i)
    {
        int x=i;
        for (int j=1;j<=prime[0]&&x>1&&prime[j]*prime[j]<=x;++j)
            while (x%prime[j]==0) --cnt[j],x/=prime[j];
        if (x>1) --cnt[num[x]];
    }
    ans1.a[0]=ans1.a[1]=1;
    for (int i=1;i<=prime[0];++i)
        while (cnt[i]) ans1=cheng(ans1,prime[i]),--cnt[i];
    for (int i=n+2;i<=n+m;++i)
    {
        int x=i;
        for (int j=1;j<=prime[0]&&x>1&&prime[j]*prime[j]<=x;++j)
            while (x%prime[j]==0) ++cnt[j],x/=prime[j];
        if (x>1) ++cnt[num[x]];
    }
    for (int i=1;i<=m-1;++i)
    {
        int x=i;
        for (int j=1;j<=prime[0]&&x>1&&prime[j]*prime[j]<=x;++j)
            while (x%prime[j]==0) --cnt[j],x/=prime[j];
        if (x>1) --cnt[num[x]];
    }
    ans2.a[0]=ans2.a[1]=1;
    for (int i=1;i<=prime[0];++i)
        while (cnt[i]) ans2=cheng(ans2,prime[i]),--cnt[i];
    ans=jian(ans1,ans2);
    for (int i=ans.a[0];i>=1;--i)
        putchar(ans.a[i]+'0');putchar('\n');
}