玲珑-动态规划

1032 - A-B

Time Limit：1s Memory Limit：128MByte

Submissions：684Solved：127

DESCRIPTION

你有$n$个球，需要把他们放到$m$个盒子里。
要求拥有最多球的盒子唯一，问方案数。

INPUT

一行两个数 $n,m$.( $n,m\le 500$)

OUTPUT

一行一个数，表示方案数。答案对 $998244353$取模。

SAMPLE INPUT

5 2

SAMPLE OUTPUT

6

SOLUTION

“玲珑杯”ACM 热身赛 # 2.5

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define mod 998244353
long long dp[550][550];
long long pre[550][550];
int main()
{
	int n,m;
	scanf("%d%d",&n,&m);
	dp[0][0]=1;
	for(int i=0;i<=n;i++) pre[0][i]=1;
	long long ans=0;
	for(int k=0;k<=n;k++)//假设最多放了k个
	{
		for(int i=1;i<=m;i++){//要放得盒子数 
			for(int j=0;j<=n;j++) {//要放得球数 
				dp[i][j]=0;//初始化为0 
				dp[i][j]=(pre[i-1][j]-(j-k>=0?pre[i-1][j-k]:0)+mod)%mod;
				pre[i][j]=((j==0?0:pre[i][j-1])+dp[i][j])%mod; //前缀和 
				//放j-1个球的情况加上放j个球的情况（累加和） 
			}
		}
		ans=(ans+m*dp[m-1][n-k])%mod;//最多的k个球有m种放法 
	} 
	printf("%lld\n",ans);
	return 0;
}