Uva-1218 Perfect Service

題目鏈接:Perfect Service

題目大意:給一顆樹,對其進行染色,使未染色的結點恰好和一個染色的結點相連,求符合題意的最少的染色數。

解題思路:此題的難點在於dp數組的第二維存儲的並不是此結點是否染色,而是當前節點和子結點的關係,關係分爲3種:

1.子結點染色
2.父結點染色,子結點不染色
3.子結點父結點均不染色

代碼如下:

#include <map>
#include <set>
#include <cmath>
#include <queue>
#include <stack>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <iomanip>
#include <iostream>
#include <algorithm>
using namespace std;
typedef long long ll;
typedef pair<int, int> P;
const int inf  = 0x3f3f3f3f;
const int maxn = 2e4 + 150;

int dp[maxn][3]; // 0 -> son : server  1 -> son : client fa : server 2 -> son : client fa : client
int ans;
vector<int> G[maxn];

int min(int a, int b, int c){
    return min(a, min(b, c));
}

void DFS(int fa, int u){
    //if(dp[u][0] != inf && dp[u][1] != inf && dp[u][2] != inf) return;

    dp[u][0] = 1; dp[u][1] = 0; dp[u][2] = 10000;
    for(int i = 0; i < G[u].size(); i++){
        int v = G[u][i];
        if(fa == v) continue;
        DFS(u, v);  
        dp[u][0] += min(dp[v][0], dp[v][1]);
        dp[u][1] += dp[v][2]; 
    }
    for(int i = 0; i < G[u].size(); i++){
        int v = G[u][i];
        if(fa == v) continue;   
        dp[u][2] = min(dp[u][2], dp[u][1] - dp[v][2] + dp[v][0]);
    }
    return;
}

int main(){
#ifdef NEKO
    freopen("Nya.txt", "r", stdin);
#endif
    ios::sync_with_stdio(false);
    cin.tie(0);
    int n;  
    while(cin >> n){
        if(n == -1) break;
        for(int i = 0; i <= n; i++) G[i].clear();
        int u, v;
        for(int i = 1; i < n; i++){ 
            cin >> u >> v;
            G[u].push_back(v);
            G[v].push_back(u);
        }
        DFS(0, 1);
        ans = min(dp[1][0], dp[1][2]);
        cin >> u;
        cout << ans << endl;
    }
    return 0;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章