POJ-3398-Perfect Service

这个题跟Uva 1218 一样，都是树形DP，题意是说在一些机器上安装服务器，使得每台计算机刚好与一台服务器相连，题目给出的计算机之间的连接关系。需要注意的是除最后一组数据以-1结尾外，其他的都以0结尾，所以只需要判断最后输入那个数是否等于-1即可

思路：这个题树形DP还是不太难，主要需要记录3个状态，我用dp[i[3]表示，其中dp[i][1]表示在i结点放置服务器至少需要多少台，dp[i][2]表示i所有孩子都不放服务器需要多少台，最麻烦的是dp[i][0]表示i位置不放置服务器，并且i的一个孩子放置了服务器最少需要多少台。那么可以得到转移方程式

dp[i][1]=min(dp[j][1],dp[j][2]);

dp[i][2]+=dp[j][0];

dp[i][0]+=dp[j][0]+min(dp[j][1]-d[j][0]);

代码：

#include<cstdio>
#include<cstring>
#include<iostream>
#include<vector>
using namespace std;
const int inf=1<<29;
const int maxn=11000;
int n,dp[maxn][3];
vector<int> g[maxn];
void Init()
{
    memset(dp,0,sizeof(dp));
    for(int i=1;i<=n;i++)
        g[i].clear();
}
void DFS(int u,int f)
{
    if(g[u].size()==1&&g[u][0]==f)
    {
        dp[u][0]=maxn*5;
        dp[u][1]=1;
        return;
    }
    int size=g[u].size(),sum=0,val=inf;
    bool is=false;
    for(int i=0;i<size;i++)
    {
        if(g[u][i]==f)
            continue;
        DFS(g[u][i],u);
        dp[u][1]+=min(dp[g[u][i]][1],dp[g[u][i]][2]);
        dp[u][2]+=dp[g[u][i]][0];
        dp[u][0]+=dp[g[u][i]][0];
        val=min(dp[g[u][i]][1]-dp[g[u][i]][0],val);
    }
    dp[u][1]++;
    dp[u][0]+=val;
}
int main()
{
    while(scanf("%d",&n)!=EOF)
    {
        Init();
        for(int i=1;i<n;i++)
        {
            int u,v;
            scanf("%d%d",&u,&v);
            g[u].push_back(v);
            g[v].push_back(u);
        }
        DFS(1,-1);
        int ans=min(dp[1][0],dp[1][1]);
        printf("%d\n",ans);
        int val;
        scanf("%d",&val);
        if(val==-1)
            break;
    }
    return 0;
}