p4463 [国家集训队] calc

分析

代码

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