题解 [CF724E] Goods transportation

Luogu Link

CF724E $O(n^2)$ 做法题解

看到题面考虑网络流。新建超级源点 $S$，超级汇点 $T$，连边：

• $(S,i,p_i),1\le i\le n$；
• $(i,T,s_i),1\le i\le n$；
• $(i,j,c),1\le i<j\le n$。

对于样例 $3$，建出的图如下所示：

然后求得 $S$ 到 $T$ 的最大流，即为答案。

---

但是这样做图上的边数是 $n^2$ 级别的，显然过不了。

怎么办呢？把最大流转为最小割，考虑 DP，设 $f_{i,j}$ 表示前 $i$ 个点中有 $j$ 个点与源点 $S$ 的边未割掉。答案即为 $\underset{i=0}{\overset{n}{\min }}{f}_{n,i}$。

接下来考虑状态转移方程。枚举当前节点 $i$，有 $j$ 个点与源点 $S$ 的边未割掉。对于节点 $i$，肯定要断掉它与 $S$ 的连边，或它与 $T$ 的连边。

• 若删除 $i$ 与 $S$ 的连边，则还需删除 $i$ 与**编号比 $i$ 小的那些节点中与 $S$ 的边未割掉的那些节点（个数为 $j$）**的连边，割掉的总边权为 $p_i+c\ast j$。
• 若删除 $i$ 与 $T$ 的连边，则只需删除边 $(i,T,s_i)$ 即可。（因为 $i$ 流到编号更大的节点再流到 $T$ 的路径这里还不需要考虑）

因此状态转移方程为：

$$f_{i,j}=\min (f_{i-1,j}+p_i+c\ast j,f_{i-1,j-1}+s_i)$$

接下来考虑边界：

枚举的时候，$i$ 肯定从 $1$ 到 $n$，$j$ 从 $0$ 到 $n$（注意 $j$ 的值为 $0$ 的时候，状态转移方程的变化，代码中体现了这一点）。

因为是求最小值，$f$ 数组最开始全设为正无穷，$f_{0,0}$ 设为 $0$。（但其实没有必要）

//CF724E
#include <bits/stdc++.h>
typedef long long ll;
using namespace std;

const int N = 1e3 + 10; 
long long c, p[N], s[N], f[N][N], ans = 1e17;
int n;

int main(){
	scanf("%d%lld", &n, &c);
	for(int i = 1; i <= n; ++ i) scanf("%lld", &p[i]);
	for(int i = 1; i <= n; ++ i) scanf("%lld", &s[i]);
	memset(f, 0x3f, sizeof(f));f[0][0] = 0;
	for(int i = 1; i <= n; ++ i){
		f[i][0] = f[i-1][0] + p[i];
		for(int j = 1; j <= i; ++ j)
			f[i][j] = min(f[i-1][j] + p[i] + c * j, f[i-1][j-1] + s[i]);
	} 
	for(int i = 0; i <= n; ++ i) ans = min(ans, f[n][i]);
	printf("%lld\n", ans);
	return 0;
}

这样空间复杂度是 $O(n^2)$ 不能通过。可以考虑滚动数组。

//CF724E
#include <bits/stdc++.h>
typedef long long ll;
using namespace std;

const int N = 1e4 + 10; 
long long c, p[N], s[N], f[2][N], ans = 1e17;
int n;

int main(){
	scanf("%d%lld", &n, &c);
	for(int i = 1; i <= n; ++ i) scanf("%lld", &p[i]);
	for(int i = 1; i <= n; ++ i) scanf("%lld", &s[i]);
	for(int i = 1; i <= n; ++ i) f[0][i] = 1e17;
	for(int i = 1; i <= n; ++ i){
		f[1][0] = f[0][0] + p[i];
		//这一句拿出来，原因见上
		for(int j = 1; j <= i; ++ j)
			f[1][j] = min(f[0][j] + p[i] + c * j, f[0][j-1] + s[i]);
		for(int j = 0; j <= i; ++ j) f[0][j] = f[1][j];
		//这里j<=i，所以f[0][i+1]后面的还是无穷大
	} 
	for(int i = 0; i <= n; ++ i) ans = min(ans, f[0][i]);
	printf("%lld\n", ans);
	return 0;
}

这里将 $f$ 数组初始为正无穷的意义是什么？可以发现在状态转移时出现了 $f_{i-1,j}$ 一项，这时 $i-1$ 有可能小于 $j$。再回想状态定义，这种情况是不合法的。如果这样初始化，这种不合法的情况就不会被考虑；反之如果初始化全为 $0$，就会出现错误。滚动数组的代码里的初始化也能起到这个效果。

所以你也可以这么初始化：

//for(int i = 1; i <= n; ++ i) f[0][i] = 1e17;
for(int i = 1; i <= n; ++ i){
	for(int j = i+1; j <= n; ++ j) f[0][j] = 1e17;
	f[1][0] = f[0][0] + p[i];
	for(int j = 1; j <= i; ++ j)
		f[1][j] = min(f[0][j] + p[i] + c * j, f[0][j-1] + s[i]);
	for(int j = 0; j <= i; ++ j) f[0][j] = f[1][j];
} 

甚至：

//for(int i = 1; i <= n; ++ i) f[0][i] = 1e17;
for(int i = 1; i <= n; ++ i){
	f[0][i+1] = 1e17;
	f[1][0] = f[0][0] + p[i];
	for(int j = 1; j <= i; ++ j)
		f[1][j] = min(f[0][j] + p[i] + c * j, f[0][j-1] + s[i]);
	for(int j = 0; j <= i; ++ j) f[0][j] = f[1][j];
}