1. 程式人生 > >uva 12093 Protecting Zonk 樹形dp

uva 12093 Protecting Zonk 樹形dp

題目大意:有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;
}