題目鏈接：UVA - 1218 Perfect Service

題意

有n臺(tái)電腦，互相以無(wú)根樹的方式連接，現(xiàn)要將其中一部分電腦作為服務(wù)器，且要求每臺(tái)電腦必須連接且只能連接一臺(tái)服務(wù)器(不包括作為服務(wù)器的電腦)，求最少需要多少臺(tái)電腦作為服務(wù)器。

思路

典型的樹形dp問(wèn)題，那么我們來(lái)建立模型。
d(u,0)：u是服務(wù)器，孩子是不是服務(wù)器均可
d(u,1)：u不是服務(wù)器，u的父親是服務(wù)器，u的孩子不能是服務(wù)器
d(u,2)：u不是服務(wù)器且u的父親不是服務(wù)器，u的孩子必須有且僅有一個(gè)是服務(wù)器。

那么顯然的
d(u,0) = 1 + Sum( Min(d(v,1), d(v,0)) ) ｜v是u的孩子
d(u,1) = Sum(d(v,2))
d(u,2)稍復(fù)雜些，因?yàn)橹荒苡幸粋€(gè)孩子是服務(wù)器，所以需要遍歷每個(gè)孩子為服務(wù)器的情況，每一輪遍歷都要同時(shí)計(jì)算其他所有孩子，需要O(N^2)的時(shí)間。
顯然這樣會(huì)有很多重復(fù)計(jì)算的情況，當(dāng)然可以記憶化來(lái)加快速度，但是還有一個(gè)更快的小技巧，因?yàn)閐(u,1)和d(u,2)的唯一區(qū)別就是d(u,2)的孩子有一個(gè)服務(wù)器
那么我們可以得出
d(u,2) = Min(d(u,1)-d(v,2)+d(v,0)) |v是u的孩子
仍是遍歷設(shè)每個(gè)孩子為服務(wù)器，每一輪的操作為O(1)，總體為O(N)

代碼

ps：因?yàn)閐(u,2)初始要設(shè)為無(wú)窮大，就設(shè)了0x3f3f3f3f，結(jié)果改成N就好了，找了好久才找出問(wèn)題出在這，因?yàn)檫@個(gè)wrong了好多遍，但還是不知道為什么會(huì)wrong，知道原因的道友請(qǐng)不吝指點(diǎn)。

第一遍a的代碼

#include <iostream> #include <algorithm> #include <cstring> #include <cstdio> #include <cstdlib> #include <vector> #include <cmath> #include <map>using namespace std;const int N = 10009; vector<int> g[N]; int dp[N][3];void dfs(int u, int fa) {for(int i=0; i<g[u].size(); i++){if(g[u][i] != fa)dfs(g[u][i], u);}dp[u][0] = 1;dp[u][1] = 0;dp[u][2] = N;for(int i=0; i<g[u].size(); i++){if(g[u][i] != fa){dp[u][0] += min(dp[g[u][i]][0], dp[g[u][i]][1]);dp[u][1] += dp[g[u][i]][2];}}bool f = true;for(int i=0; i<g[u].size(); i++){if(g[u][i] != fa){f = false;dp[u][2] = min(dp[u][2], dp[u][1]+dp[g[u][i]][0]-dp[g[u][i]][2]);}} }int main() {int n;while(cin>>n){memset(dp, -1, sizeof(dp));int a, b;for(int i=1; i<n; i++){cin>>a>>b;g[a].push_back(b);g[b].push_back(a);}dfs(1, -1);cout<<min(dp[1][0], dp[1][2])<<endl;cin>>a;if(a == -1)break;for(int i=1; i<=n; i++)g[i].clear();}return 0; }

觀摩大神代碼后進(jìn)行修改的精簡(jiǎn)版

#include <iostream> #include <algorithm> #include <cstring> #include <cstdio> #include <cstdlib> #include <vector> #include <cmath> #include <map>using namespace std;const int N = 10009; vector<int> g[N]; int dp[N][3];void dfs(int u, int fa) {dp[u][0] = 1;dp[u][1] = 0;dp[u][2] = N;for(int i=0; i<g[u].size(); i++){if(g[u][i] != fa){dfs(g[u][i], u);dp[u][0] += min(dp[g[u][i]][0], dp[g[u][i]][1]);dp[u][1] += dp[g[u][i]][2];}}for(int i=0; i<g[u].size(); i++){if(g[u][i] != fa)dp[u][2] = min(dp[u][2], dp[u][1]+dp[g[u][i]][0]-dp[g[u][i]][2]);} }int main() {int n;while(cin>>n){int a, b;for(int i=1; i<n; i++){cin>>a>>b;g[a].push_back(b);g[b].push_back(a);}dfs(1, -1);cout<<min(dp[1][0], dp[1][2])<<endl;cin>>a;if(a == -1)break;for(int i=1; i<=n; i++)g[i].clear();}return 0; }

總結(jié)

以上是生活随笔為你收集整理的UVA - 1218 Perfect Service(树形dp)的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。