【发布时间】:2015-08-14 12:33:23
【问题描述】:
假设我们有一个加权树和该树的一些顶点集。问题是通过从树中删除边来隔离这些顶点(例如它们之间不应该有路径),从而使删除边的权重之和最小。
我一直在尝试解决这个问题大约两个小时,然后我在 C++ 中找到了解决方案,但没有任何解释。据我了解,它使用动态编程技术。
我的问题是解决这个问题的基本思路是什么?
谢谢!
这里是解决方案。
#include <iostream>
#include <cstring>
#include <vector>
using namespace std;
#define PB push_back
#define MP make_pair
typedef long long LL;
const int MAXN = 100005;
const LL inf = 1LL << 55;
int N, K;
bool bad[MAXN];
LL dp[MAXN][2];
vector<pair<int, int> > adj[MAXN];
void dfs(int u, int fa) {
dp[u][1] = 0;
dp[u][0] = bad[u] ? inf : 0;
for (int i = 0; i < adj[u].size(); i++) {
int v = adj[u][i].first, w = adj[u][i].second;
if (v != fa) {
dfs(v, u);
dp[u][1] += min(dp[v][0], dp[v][1] + w);
dp[u][1] = min(dp[u][1], dp[u][0] + dp[v][1]);
if (!bad[u])
dp[u][0] += min(dp[v][0], dp[v][1] + w);
}
}
}
int main() {
cin >> N >> K;
for (int i = 1; i < N; i++) {
int a, b, c;
cin >> a >> b >> c;
adj[a].PB(MP(b, c)); adj[b].PB(MP(a, c));
}
memset(bad, false, sizeof(bad));
for (int i = 0; i < K; i++) {
int x;
cin >> x;
bad[x] = true;
}
dfs(0, -1);
LL ret = min(dp[0][0], dp[0][1]);
cout << ret << endl;
return 0;
}
【问题讨论】:
-
这真是糟糕的 C++ 代码..
-
能否请您也发布确切的问题说明? “隔离那些顶点”到底是什么意思?应该没有从根到它们的路径,或者我们应该有一个连接的组件(或更多?),其中只有孤立的顶点(它们)?
-
你所说的树,是指二叉树、N叉树还是“通用树”(有根连通无环图)?二叉树会(编辑:可能)大大简化问题(我仍然难以理解您发布的代码的意图,变量的命名真的很糟糕)。
-
一般树。它是有竞争力的程序员的代码。这就是为什么它的名字如此糟糕的原因:)
标签: c++ algorithm dynamic-programming