文章目录
E. Arena
题意:
现有
n
n
n位勇士,生命值在
1
∼
x
1\sim x
1∼x ,每回合所有勇士减少
t
−
1
t-1
t−1点生命值,其中
t
t
t是生命值为正的勇士数量。问:有多少种分配各勇士生命值的方法,可以使得最终所有勇士生命值均降为
0
0
0及以下。
思路:DP
状态表示:
设
d
p
[
i
]
[
j
]
dp[i][j]
dp[i][j]表示
i
i
i个勇士,其中勇士血量最多为
j
j
j,经过若干次后所有勇士血量
≤
0
\le0
≤0的方案数
状态转移:
如果
i
−
1
>
=
j
,
d
p
[
i
]
[
j
]
=
j
i
i-1>=j,dp[i][j]=j^i
i−1>=j,dp[i][j]=ji
否则
- d p [ i ] [ j ] + = d p [ k ] [ j − i + 1 ] ∗ ( i − 1 ) i − k ∗ C ( i , k ) dp[i][j]+=dp[k][j-i+1]*(i-1)^{i-k}*C(i,k) dp[i][j]+=dp[k][j−i+1]∗(i−1)i−k∗C(i,k),枚举 k k k,表示第一次有 k k k个勇士死亡,这 k k k个勇士每个勇士的血量可以取 1 ∼ i − 1 1\sim i-1 1∼i−1,再乘于 C ( n , k ) C(n,k) C(n,k)表示从 n n n个勇士中取 k k k个
- d p [ i ] [ j ] + = ( i − 1 ) i dp[i][j]+=(i-1)^i dp[i][j]+=(i−1)i,表示 i i i个勇士全死的方案
答案: d p [ n ] [ x ] dp[n][x] dp[n][x]
#include <bits/stdc++.h>
using namespace std;
#define rep(i,a,b) for(int i=(a);i<=(b);++i)
#define per(i,a,b) for(int i=(a);i>=(b);--i)
#define ll long long
const int N=510,M=2*N;
const int inf=0x3f3f3f3f;
const ll INF=0x3f3f3f3f3f3f3f3f;
const ll mod=998244353;
ll qpow(ll a,ll b){
ll res=1;
while(b){
if(b&1) res=res*a%mod;
b>>=1;
a=a*a%mod;
}
return res%mod;
}
ll C[N][N];
void init(){
C[0][0]=1;
rep(i,1,505){
rep(j,0,i){
if(!j) C[i][j]=1;
else C[i][j]=(C[i-1][j]+C[i-1][j-1])%mod;
}
}
}
ll dp[N][N]; //dp[i][j]表示i个勇士,最多血量为j
int n,x;
void solve(){
init();
scanf("%d%d",&n,&x);
rep(i,2,n)
rep(j,1,x){
if(i-1>=j) dp[i][j]=qpow(j,i);
else{
rep(k,2,i){
(dp[i][j]+=dp[k][j-i+1]*qpow(i-1,i-k)%mod*C[i][k])%=mod;
}
(dp[i][j]+=qpow(i-1,i))%=mod;
}
}
printf("%lld\n",dp[n][x]);
}
signed main(){
//int _;scanf("%d",&_);while(_--)
solve();
return 0;
}