【JZOJ6218】【20190615】卖弱
题目
题解
我写的另一种方法,复杂度是\(O(Tm+nm)\)的,这是huangzhaojun写的题解。。。
#include<cstring>
#include<cstdio>
#define N 210
#define mo 998244353
using namespace std;
int n,a,dp[N][N],A=200,fr[N];
int qpow(int a,int i){
int r=1;for(;i;i>>=1,a=1ll*a*a%mo)if(i&1)r=1ll*r*a%mo;return r;
}
int main(){
freopen("sell.in","r",stdin);
freopen("sell.out","w",stdout);
int _;scanf("%d",&_);
dp[0][0]=1;fr[0]=1;
for(int i=1;i<=A;++i)fr[i]=1ll*fr[i-1]*i%mo;
for(int i=1;i<=A;++i)for(int j=0;j<=i;++j)dp[i][j]=((j?1ll*dp[i-1][j-1]*(i-j+1):0)+dp[i-1][j])%mo;
while(_--){
int ans=0;
scanf("%d %d",&n,&a);
for(int i=0;i<=a;++i)ans=(ans+1ll*(i%2==1?-1:1)*dp[a][i]*qpow(a-i,n-a)%mo*fr[a-i])%mo;
printf("%d\n",(ans+mo)%mo);
}
}