Atcoder DP Contest M题解

atcoder 题目传送门，经典的 dp + 前缀和优化。

更好的阅读体验。

---

题意

有 $n$ 个不同的小朋友和 $k$ 个相同的糖，给每个小朋友分配若干个糖且糖要分配完，每个小朋友分配糖的上限个数为 $a_i$，求分配方案总数。

• $1\le n\le 100$

• $1\le k\le 10^5$

• $1\le a_i\le k$

---

思路

• 朴素的 dp 状态：$dp(i,j)$ 表示考虑到第 $i$ 个小朋友，当前已经用掉 $j$ 个糖的方案数。

• 转移方程：
  $$dp(i,j)=\sum _{k=\max (0,j-a_{i-1})}^{j}dp(i-1,k)$$

这样一来转移的复杂度是 $O(k)$，总复杂度为 $O(n{k}^2)$，会超时。这时候我们进行优化，就必须要观察转移方程的特点了。

• 优化：发现公式里每个 dp 值都由上一行中一段连续的 dp 值之和转移而来，可以前缀和优化转移，每次用 $sum$ 数组预处理上一行 $dp$ 的前缀和，可以 $O(1)$ 地求出连续一段 dp 值之和并转移。具体处理如下：
  $$sum(i)=\sum _{j=0}^{i}dp(i-1,j)$$

时间复杂度：$O(nk)$

空间复杂度：$O(nk)$

---

代码

#include <bits/stdc++.h>
#define ll long long
#define pb push_back
#define pii pair<int,int>
#define mp make_pair
#define F first
#define S second
using namespace std;
const ll MOD=1e9+7;
int n,m,a[105];
ll dp[105][100005],sum[100005];
ll getsum(int l,int r)//前缀和
{
	return (l==0?sum[r]:(sum[r]+MOD-sum[l-1])%MOD);
}
int main()
{
	cin>>n>>m;
	for(int i=0;i<n;i++)
		cin>>a[i];
	dp[0][0]=1ll;
	for(int i=1;i<=n;i++)
	{
		sum[0]=dp[i-1][0];//处理上一行dp的前缀和
			for(int k=1;k<=m;k++)
				sum[k]=(sum[k-1]+dp[i-1][k])%MOD;
		for(int j=0;j<=m;j++) //按公式转移
			dp[i][j]=getsum(max(0,j-a[i-1]),j);
	}
	cout<<dp[n][m]<<endl;
	return 0;
}

其实我的 dp 公式是从前面的值转移到当前值的。也有其他的选手喜欢用当前值更新后面值的转移方式。这样的话优化就会运用到一个差分数组，其实和前缀和是异曲同工的。

希望题解对你有帮助！