「HNOI/AHOI2018」道路

传送门

Luogu

解题思路

考虑树形 \(\text{DP}\)

设状态 \(dp[u][i][j]\) 表示从首都走到点 \(u\) ，经过 \(i\) 条公路，\(j\) 条铁路的最小不方便值。

对于叶子节点也就是村庄，直接初始化，对于非叶子节点也就是城市，就从两个儿子向上转移。

而现在的一个问题就是：我们的 \(dp\) 数组根本开不下。

我们不得不优化空间。

考虑到一个性质：每次的答案都是从儿子向上合并，合并过后的节点就没用了。

所以我们只需要在dfs的过程中给每个节点记一个 \(dfn\)，满足：

\(dfn_{ls}=dfn_{fa} + 1, dfn_{rs}=dfn_{fa} + 2\)

我们把空间回收利用起来，很容易计算 \(dp\) 的第一位，只需要开到 \(2*40+1=81\) 即可。

这样就可以愉快的 DP 了，答案就是 \(dp[1][0][0]\)

细节注意事项

• 咕咕咕

参考代码

#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdlib>
#include <cstdio>
#include <cctype>
#include <cmath>
#include <ctime>
#define rg register
using namespace std;
template < typename T > inline void read(T& s) {
	s = 0; int f = 0; char c = getchar();
	while (!isdigit(c)) f |= c == '-', c = getchar();
	while (isdigit(c)) s = s * 10 + c - 48, c = getchar();
	s = f ? -s : s;
}

typedef long long LL;
const int _ = 20010;

int n, a[_ << 1], b[_ << 1], c[_ << 1], ch[2][_];
LL dp[90][50][50];

inline void dfs(int s, int u, int x, int y) {
	if (u >= n) {
		for (rg int i = 0; i <= x; ++i)
			for (rg int j = 0; j <= y; ++j)
				dp[s][i][j] = 1ll * c[u] * (a[u] + i) * (b[u] + j);
		return;
	}
	dfs(s + 1, ch[0][u], x + 1, y);
	dfs(s + 2, ch[1][u], x, y + 1);
	int ls = s + 1, rs = s + 2;
	for (rg int i = 0; i <= x; ++i)
		for (rg int j = 0; j <= y; ++j)
			dp[s][i][j] = min(dp[ls][i + 1][j] + dp[rs][i][j], dp[ls][i][j] + dp[rs][i][j + 1]);
}

int main() {
#ifndef ONLINE_JUDGE
	freopen("in.in", "r", stdin);
#endif
	read(n);
	for (rg int i = 1; i < n; ++i) {
		read(ch[0][i]), ch[0][i] = ch[0][i] < 0 ? n - ch[0][i] - 1 : ch[0][i];
		read(ch[1][i]), ch[1][i] = ch[1][i] < 0 ? n - ch[1][i] - 1 : ch[1][i];
	}
	for (rg int i = 1; i <= n; ++i)
		read(a[i + n - 1]), read(b[i + n - 1]), read(c[i + n - 1]);
	dfs(1, 1, 0, 0);
	printf("%lld\n", dp[1][0][0]);
	return 0;
}

完结撒花 \(qwq\)