Educational Codeforces Round 116 (Rated for Div. 2)

---

E. Arena

题意：
现有$n$位勇士，生命值在$1\sim x$ ，每回合所有勇士减少$t-1$点生命值，其中$t$是生命值为正的勇士数量。问：有多少种分配各勇士生命值的方法，可以使得最终所有勇士生命值均降为$0$及以下。

思路：DP
状态表示：
设$dp[i][j]$表示$i$个勇士，其中勇士血量最多为$j$，经过若干次后所有勇士血量$\le 0$的方案数
状态转移：
如果$i-1>=j,dp[i][j]=j^i$
否则

1. $dp[i][j]+=dp[k][j-i+1]\ast (i-1{)}^{i-k}\ast C(i,k)$，枚举$k$，表示第一次有$k$个勇士死亡，这$k$个勇士每个勇士的血量可以取$1\sim i-1$,再乘于$C(n,k)$表示从$n$个勇士中取$k$个
2. $dp[i][j]+=(i-1{)}^i$，表示$i$个勇士全死的方案

答案：$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;
}