題目鏈接: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;
}