http://acm.hdu.edu.cn/showproblem.php?pid=3944
思路看https://blog.csdn.net/clover_hxy/article/details/55101759
这题逆元用的时候再算300+ms,如果提前全部预处理好,就3000+ms超时了。
还要注意开数组不要1w*1w,因为1w以内的素数只有1200+个,节省空间。
#include<bits/stdc++.h>
using namespace std;
#define ll long long
int n,m,p,Case;
int idx[10000+100],prime[1500],fac[10000+100][1500],inv[10000+100][1500];
bool vis[10000+100];
int pow_mod(int a,int n,int m)
{
if(!n)return 1;
int x=pow_mod(a,n/2,m);
ll ans=(ll)x*x%m;
if(n&1)ans=ans*a%m;
return (int)ans;
}
void init()
{
for(int i=2;i<=100;i++)if(!vis[i])
for(int j=i*i;j<=10000;j+=i)vis[j]=1;
int cnt=0;
for(int i=2;i<=10000;i++)if(!vis[i])idx[i]=++cnt,prime[cnt]=i;
for(int j=0;j<=cnt;j++)fac[0][j]=1;
for(int i=1;i<=10000;i++)for(int j=1;j<=cnt;j++)fac[i][j]=fac[i-1][j]*i%prime[j];
// for(int j=0;j<=cnt;j++)inv[0][j]=1;
// for(int i=1;i<=10000;i++)for(int j=1;j<=cnt;j++)inv[i][j]=pow_mod(fac[i][j],prime[j]-2,prime[j]);
}
int Inv(int i,int j){return pow_mod(fac[i][j],prime[j]-2,prime[j]);}
int C(int n,int m)
{
if(m>n)return 0;
return fac[n][idx[p]]*Inv(m,idx[p])%p*Inv(n-m,idx[p])%p;
}
int Lucas(int n,int m)
{
if(m==0)return 1;
return (long long)C(n%p,m%p)*Lucas(n/p,m/p)%p;
}
int main()
{
freopen("input.in","r",stdin);
init();
while(~scanf("%d%d%d",&n,&m,&p))
{
if(m<=n/2)printf("Case #%d: %d\n",++Case,(Lucas(n+1,m)+n-m)%p);
else printf("Case #%d: %d\n",++Case,(Lucas(n+1,m+1)+m)%p);
}
return 0;
}