cf1204E - Natasha, Sasha and the Prefix Sums

题目链接

题意：由n个1和m个-1的构成所有序列中，Σf(前缀和最大值)是多少。如果当前序列的前缀和最大值小于0，则认为是0.(n,m<=2000,最后结果对998244853取模）

题解：首先看数据量明显是个二维的dp问题，dp[i][j] 表示答案，那么如何转移呢？首先若求出dp[i-1][j] 的答案，那么往前面加上一个1就相当于将dp[i-1][j]的所有序列的贡献都加上了1，所以前半部分是dp[i-1][j] + $\left(\genfrac{}{}{0ex}{}{i+j-1}{j}\right)$,若从dp[i][j-1] 转移过来，那么相当于贡献减去由（i , j-1）构成的序列的个数。但是由于变成负数现当于变成0，所以此时需要加上由（i , j-1）构成的序列中，满足题意的，最大前缀和为0的个数（负数也算0）,设这个数为k[i][j] 这个数怎么求，若i==0明显看出k[i][j] =1，若 i>j ,k[i][j] =0。否则k[i][j]=k[i-1][j]+k[i][j-1] 因为相当于这两种情况下后面填1或者-1，又由于i<=j时k值才存在，所以最后总的前缀和是<=0的，往后面添加1和-1的情况一定满足要求。最后的dp方程为：

dp[i][j]=dp[i-1][j]+ $\left(\genfrac{}{}{0ex}{}{i+j-1}{j}\right)$+dp[i][j-1]-( $\left(\genfrac{}{}{0ex}{}{i+j-1}{i}\right)$-k[i][j-1])

总结一下：遇到n方dp一般考虑往后放单位来转移状态，但是也可以往前放单位来转移~

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int mod = 998244853;
const int maxn = 2005;
int n,m;
int dp[maxn][maxn];
int k[maxn][maxn];//表示n个1与m个-1能组成最大f(a)值为0的情况
int C[2*maxn][2*maxn];
void init() {
	for(int i=0; i<=4000; i++)C[0][i]=1;
	for(int i=1; i<=4000; i++) {
		for(int j=i; j<=4000; j++) {
			C[i][j]=((ll)C[i][j-1]+C[i-1][j-1])%mod;
		}
	}
}
int main() {
	init();
	scanf("%d%d",&n,&m);
	for(int i=0; i<=n; i++) {
		for(int j=0; j<=m; j++) {
			if(i==0) {
				k[i][j]=1;
			} else if(i>j) {
				k[i][j]=0;
			} else {
				k[i][j]=(k[i][j-1]+k[i-1][j])%mod;
			}
		}
	}
	for(int i=1;i<=n;i++)dp[i][0]=i;
	for(int i=1;i<=m;i++)dp[0][i]=0;
	for(int i=1;i<=n;i++){
		for(int j=1;j<=m;j++){
			dp[i][j]=((ll)dp[i-1][j]+C[j][i+j-1]+dp[i][j-1]-(C[i][j+i-1]-k[i][j-1]))%mod;
			dp[i][j]=(dp[i][j]+mod)%mod;
		}
	}
	printf("%d\n",dp[n][m]);
	return 0;
}