p4463 [国家集训队] calc

分析

 代码

#include<bits/stdc++.h>
using namespace std;
int n,m,A,mod,dp[510][1010],Ans;
inline int pw(int x,int p){
    int res=1;
    while(p){
      if(p&1)res=1ll*res*x%mod;
      x=1ll*x*x%mod;
      p>>=1;
    }
    return res;
}
int main(){
    int i,j,k;
    scanf("%d%d%d",&A,&n,&mod);
    m=2*n+1;
    for(i=0;i<=m;i++)dp[0][i]=1;
    for(i=1;i<=n;i++)
      for(j=1;j<=m;j++)
        dp[i][j]=(dp[i][j-1]+1ll*dp[i-1][j-1]*j%mod)%mod;
    int t=1;
    for(i=1;i<=n;i++)t=1ll*t*i%mod;
    if(A<=m){
      printf("%d\n",1ll*dp[n][A]*t%mod);
      return 0;
    }
    for(i=0;i<=m;i++){
      int res=1,p=1,q=1;
      for(j=0;j<=m;j++)if(j!=i)res=1ll*res*(A-j)%mod*pw(i-j,mod-2)%mod;
      res=1ll*res*dp[n][i]%mod;
      Ans=((Ans+res)%mod+mod)%mod;
    }
    printf("%d\n",1ll*Ans*t%mod);
    return 0;
}