uva 12093 Protecting Zonk 樹形dp
阿新 • • 發佈:2019-01-08
題目大意:有n個城市和n-1條路組成了一個樹,現在有花費為c1的機器人A和花費為c2的機器人B,兩種機器人都是無限量的,如果在城市u放置機器人A,此時與u連線的邊都會被覆蓋,如果在城市u放置機器人B,那麼與u相連的邊都會被覆蓋,且與u相連的點所相連的邊也會被覆蓋,現問將所有道路都覆蓋所需最小花費
dp[u][0] :以點u為根的子樹下的邊全部被覆蓋,且沒有向u節點上方覆蓋
dp[u][1]:以點u為根的子樹下的邊全部覆蓋,且向上覆蓋長度為1
dp[u][2]:以點u為根的子樹下的邊全部覆蓋,且向上覆蓋長度為2
dp[u][3]:以點u為根的子樹的子樹裡的邊都被覆蓋,但是u和子樹間的邊不一定被覆蓋
#include <cstdio> #include <algorithm> #include <vector> using namespace std; int n, c1, c2; const int maxn = 1e4 + 10; const int INF = 0x3f3f3f3f; vector<int> G[maxn]; int dp[maxn][4]; void dfs(int u, int fa) { dp[u][0] = dp[u][3] = 0; dp[u][1] = c1; dp[u][2] = c2; int min_A = INF, sum = 0; for (int i = 0; i < G[u].size(); i++) { int v = G[u][i]; if (v == fa) continue; dfs(v, u); int min_of_t = min(min(dp[v][0], dp[v][1]), dp[v][2]); dp[u][0] += dp[v][1]; dp[u][1] += min_of_t; dp[u][2] += min(min_of_t, dp[v][3]); dp[u][3] += min_of_t; sum += min_of_t; min_A = min(min_A, dp[v][2] - min_of_t); } sum += min_A; dp[u][1] = min(dp[u][1], sum); } int main() { while (scanf("%d%d%d", &n, &c1, &c2) != EOF) { if (n == 0) break; for (int i = 1; i <= n; i++) G[i].clear(); int u, v; for (int i = 1; i < n; i++) { scanf("%d%d",&u, &v); G[u].push_back(v); G[v].push_back(u); } dfs(1, -1); int res = min(min(dp[1][0], dp[1][1]), dp[1][2]); printf("%d\n", res); } return 0; }