uva 1218 完美服务器 树形dp 染色问题

详解见紫书 283页。

<pre name="code" class="cpp">#include<bits/stdc++.h>
#define INF 10000000
using namespace std;
const int N= 10010;
vector<int>vec[N];
int dp[N][3];  ///
int vis[N];
void dfs(int u){
    int len=vec[u].size();
    vis[u]=1;
    dp[u][0]=1;
    dp[u][1]=0;
    dp[u][2]=N;  ///初始化太大了会wa,会出现累加到了int上限值然后爆掉
    
    queue<int>q;  ///普通队列，记录没访问的子节点
    for(int i=0;i<len;i++){
        int v=vec[u][i];
        if(vis[v])continue;
        dfs(v);
        dp[u][0]+=min(dp[v][0],dp[v][1]);
        dp[u][1]+=dp[v][2];
        q.push(v);
    }
    ///因为dp[u][1]必须是算完后的结果
    while(!q.empty()){
        int v=q.front();  ///一定是u的未访问的子节点
        dp[u][2]=min(dp[u][1]-dp[v][2]+dp[v][0],dp[u][2]);
        q.pop();
    }
}
int main(){
    int n,op,u,v;
    while(~scanf("%d",&n)&&n!=-1){
        if(!n) continue;
        memset(vis,0,sizeof(vis));
        memset(vec,0,sizeof(vec));
        for(int i=0;i<n-1;i++){
            scanf("%d%d",&u,&v);
            vec[u].push_back(v);
            vec[v].push_back(u);
        }
        dfs(1);
        printf("%d\n",min(dp[1][0],dp[1][2]));  ///不能包含dp[1][1];
    }
    return 0;
}