codeforces 543A Writing Code (dp)

一开始想到的dp: f [ i ][ j ][ k ]表示前i个人打了j行有k个错误的方案数,那么转移方程为:f [ i ][ j ][ k ]=f [ i-1][ j-h ][ k-a[i]*h ] (0<=h<=j)

很明显超时+MLE。

这时会发现转移只与f [ i-1 ]有关,可以通过降一维来优化,删去第一维。

f [ j ][ k ]=f [ j ][ k ]+f [ j-1 ][ k-a[i] ]。 这时的f [ j ][ k ]其实是i-1是的f [ j ][ k ]。

#include<stdio.h>
#include<algorithm>
using namespace std;
int n,m,b,mod;
int a[550];
int f[505][505],g[505][505];
int main()
{
	scanf("%d %d %d %d",&n,&m,&b,&mod);
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&a[i]);
	}
	f[0][0]=1;
	for(int i=1;i<=n;i++)
	{
		for(int j=1;j<=m;j++)
		{
			for(int k=a[i];k<=b;k++)
			{
				f[j][k]=(f[j][k]+f[j-1][k-a[i]])%mod;
			}
		}
	}
	int ans=0;
	for(int i=0;i<=b;i++)
	{
		ans=(f[m][i]+ans)%mod;
	}
	printf("%d\n",ans);
}


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章