HDU - 3944 DP?（预处理+Lucas定理）

传送门

---

首先不难发现只有两种走法是最划算的：

• 对于第一种先直走后斜着的走法，显然是如下的一个公式：

  $C(n,k)+C(n-1,k-1)+...+C(n-k+1,1)+C(n-k,0)+n-k$

  根据公式$C(n,k)=C(n-1,k)+C(n-1,k-1)$

  我们将最后一项$C(n-k,0)$换成$C(n-k+1,0)$，显然可以和倒数第二项合并得到$C(n-k+2,1)$，那么又可以和倒数第三项合并…，最后得到一个计算公式$C(n+1,k)+n-k$

• 对于第二种先斜着走后直走的走法，显然是如下的公式：

  $C(n,k)+C(n-1,k)+...+C(k,k)+k$

  同理最后得到的是一个如下的公式：$C(n+1,k+1)+k$

观察不难得知当$k\le \frac{2}{n}$时，选择第一种方案更优；否则选择第二种方案更优

当我们直接计算时，发现实际的复杂度可能最坏为$T\ast (lo{g}_{2}n{)}^2$，因此就超时了。观察可知卢卡斯定理每次计算的组合数的$n,m$都小于$p$，那么可以预处理保存$p$的范围内的阶乘对$p$取模的结果，此时的时间复杂度为$O(T\ast logn)$

//
// Created by Happig on 2020/8/20
//
#include <bits/stdc++.h>
#include <unordered_map>
#include <unordered_set>
using namespace std;
#define fi first
#define se second
#define pb push_back
#define ins insert
#define Vector Point
#define lowbit(x) (x&(-x))
#define mkp(x, y) make_pair(x,y)
#define mem(a, x) memset(a,x,sizeof a);
typedef long long ll;
typedef long double ld;
typedef unsigned long long ull;
typedef pair<int, int> pii;
typedef pair<ll, ll> pll;
typedef pair<double, double> pdd;
const double eps = 1e-8;
const double pi = acos(-1.0);
const int inf = 0x3f3f3f3f;
const double dinf = 1e300;
const ll INF = 1e18;
const int Mod = 1e9 + 7;
const int maxn = 1e4 + 10;

vector<int> prime;
bool is_prime[maxn];
unordered_map<int,int> mp;
ll fac[maxn][1300];
int num;

void getPrime(){  //欧拉筛求素数
    int m=sqrt(maxn+0.5);
    memset(is_prime,1,sizeof is_prime);
    is_prime[0]=is_prime[1]=0;
    for(int i=2;i<maxn;i++){
        if(is_prime[i]) prime.push_back(i);
        for(int j=0;j<prime.size() && 1LL*i*prime[j]<maxn;j++){
            is_prime[i*prime[j]]=0;
            if(i%prime[j]==0) break;
        }
    }
    //cout<<prime.size()<<endl;
    mp.clear(),num=0;
    for(auto i: prime) mp[i]=++num;
}

ll qkp(ll x,ll n,ll p){
    ll ans=1;
    while(n){
        if(n&1) ans=ans*x%p;
        x=x*x%p;
        n>>=1;
    }
    return ans;
}

ll inv(ll x,ll p){  //求逆元
    return qkp(x,p-2,p);
}

ll cal(ll n,ll m,ll p){
    if(m>n) return 0;
    return fac[n][mp[p]]*inv(fac[m][mp[p]],p)%p*inv(fac[n-m][mp[p]],p)%p;
}

ll lucas(ll n,ll m,ll p){
    if(!m) return 1;
    return cal(n%p,m%p,p)*lucas(n/p,m/p,p)%p;
}

void init(){
    for(auto p: prime){
        int j=mp[p];
        fac[0][j]=1;
        for(int i=1;i<maxn;i++)
            fac[i][j]=fac[i-1][j]*i%p;
    }
}

int main() {
    //freopen("in.txt","r",stdin);
    //freopen("out.txt","w",stdout);
    //ios_base::sync_with_stdio(0),cin.tie(0),cout.tie(0);
    int kase=0;
    ll n,m,p;
    getPrime();
    init();
    while(scanf("%lld%lld%lld",&n,&m,&p)!=EOF){
        if(m<=n/2) printf("Case #%d: %lld\n",++kase,(lucas(n+1,m,p)+(n-m+p)%p)%p);
        else printf("Case #%d: %lld\n",++kase,(lucas(n+1,m+1,p)+m)%p);
    }
    return 0;
}