bzoj 4856 [Jsoi2016]病毒感染 区间dp

题面

题目传送门

解法

• 首先显然可以发现，某一次回头一定会把之前没有被救过的村庄全都救一遍，那么最后的路径一定是不断向前走然后回头再往前走，直到所有村庄全都被救为止。
• 那么我们可以设$f[i]$表示编号为$1-i$的村庄全部被救最少的死亡人数。
• 对于转移，可以枚举这一次回头后到达的最后一个村庄，即这一轮最早一开始经过但没有救的村庄的编号$j$，那么便有如下转移：
• $f[i]=min(f[j]+g[j][i]+(s[n]-s[i])\ast [(i-j)\ast 3+(i-j+1)+1])$，其中$g[j][i]$表示在从$j$到$i$再到$j$最后返回$i$的过程中最少的死亡人数，且$j$一开始被经过时并不会救。
• 那么考虑如何求解$g[i][j]$。我们可以设$h[i][j]$表示在从抵达$i$这个村庄到救完$[i,j]$中的所有村庄并回到$j$这段时间中最少的死亡人数。
• 考虑$i$这个点是否在经过时就被救，那么有转移$h[i][j]=h[i+1][j]+min(s[j]-s[i]+a[i]\ast (j-i)\ast 3,(s[j]-s[i])\ast 2)$。
• 求解完$h[i][j]$后，那可以自然地求出$g[i][j]=h[i+1][j]+s[j]-s[i]+a[i]\ast 3\ast (j-i)$。
• 时间复杂度：$O(n^2)$。

代码

#include <bits/stdc++.h>
#define ll long long
using namespace std;
template <typename T> void chkmax(T &x, T y) {x = x > y ? x : y;}
template <typename T> void chkmin(T &x, T y) {x = x < y ? x : y;}
template <typename T> void read(T &x) {
	x = 0; int f = 1; char c = getchar();
	while (!isdigit(c)) {if (c == '-') f = -1; c = getchar();}
	while (isdigit(c)) x = x * 10 + c - '0', c = getchar(); x *= f;
}
const int N = 3010;
ll a[N], f[N], s[N], g[N][N], h[N][N];
int main() {
	int n; read(n);
	for (int i = 1; i <= n; i++) read(a[i]), s[i] = s[i - 1] + a[i];
	for (int l = 1; l <= n; l++)
		for (int i = 1; i <= n - l + 1; i++) {
			int j = i + l - 1;
			h[i][j] = h[i + 1][j] + min(s[j] - s[i] + a[i] * 3 * (j - i), (s[j] - s[i]) * 2);
		}
	for (int l = 1; l <= n; l++)
		for (int i = 1; i <= n - l + 1; i++) {
			int j = i + l - 1;
			g[i][j] = h[i + 1][j] + s[j] - s[i] + a[i] * 3 * (j - i);
		}
	memset(f, 0x3f, sizeof(f)); f[0] = 0;
	for (int i = 1; i <= n; i++)
		for (int j = 1; j <= i; j++)
			chkmin(f[i], f[j - 1] + g[j][i] + (s[n] - s[i]) * ((i - j) * 3 + i - j + 2));
	cout << f[n] << '\n';
	return 0;
}