Yahoo Programming Contest 2019.F.Pass(DP)

题目链接


惊了这是什么F题...怎么我都能做出来...以后atcoder的比赛也不能走神了万一有个这样的F呢(CF已有多次了= =)

(f[i][j])表示Takahashi现在一共有(i)个球,其中(j)个红球(i-j)个蓝球的方案数。
当一共有(i)个球时,能得到的最多的红球数就是(1simmin(i,n))中所有人的红球数之和,蓝球同理。
转移时限制一下当前红蓝球数不超过最多可得球数即可(一定存在可行方案)。


//34ms	60416KB
#include <cstdio>
#include <cstring>
#include <algorithm>
#define gc() getchar()
#define mod 998244353
#define Add(x,v) (x+=v)>=mod&&(x-=mod)
typedef long long LL;
const int N=4005;

int f[N][N],sum1[N],sum2[N];
char s[N];

int main()
{
	scanf("%s",s+1);
	int n=strlen(s+1);
	for(int i=1; i<=n; ++i)
	{
		if(s[i]=='0') sum1[i]=2;
		else if(s[i]=='1') sum1[i]=sum2[i]=1;
		else sum2[i]=2;
		sum1[i]+=sum1[i-1], sum2[i]+=sum2[i-1];
	}
	f[0][0]=1;
	for(int i=1,lim=n<<1; i<=lim; ++i)
	{
		int n1=sum1[std::min(i,n)],n2=sum2[std::min(i,n)];
		for(int j=0; j<=i; ++j)
			if(n1>=j && n2>=i-j)
			{
				f[i][j]=f[i-1][j];//Add
				if(j) Add(f[i][j],f[i-1][j-1]);
			}
	}
	printf("%d
",f[n<<1][sum1[n]]);

	return 0;
}
原文地址:https://www.cnblogs.com/SovietPower/p/10389540.html