【树形dp】UVA1218Perfect Service

 

注意这道题目inf开太大会爆掉int！！

 

代码

 

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cmath>
#include<cstring>
using namespace std;
const int maxn=1e5+5;
const int inf=maxn;
int n,tot,dp[maxn][3],head[maxn];
struct edge
{
	int to,nxt;
}e[maxn<<1];
void add(int x,int y)
{
	e[++tot].nxt=head[x];
	e[tot].to=y;
	head[x]=tot;
}
void dfs(int u,int fa)
{
	dp[u][0]=1; dp[u][1]=0; dp[u][2]=inf;
	for(int i=head[u];i;i=e[i].nxt)
	{
		int to=e[i].to;
		if(to==fa) continue;
		dfs(to,u);
		dp[u][0]+=min(dp[to][0],dp[to][1]);
		dp[u][1]+=dp[to][2];
	}
	for(int i=head[u];i;i=e[i].nxt)
	{
		int to=e[i].to;
		if(to==fa) continue;
		dp[u][2]=min(dp[u][2],dp[u][1]+dp[to][0]-dp[to][2]);
	}
}
int main()
{
	freopen("a.in","r",stdin);
	freopen("a.out","w",stdout);
	int x,y;
	while(n!=-1 && scanf("%d",&n))
	{
		tot=0;
		memset(head,0,sizeof(head));
		for(int i=1;i<n;i++)
		{
			scanf("%d%d",&x,&y);
			add(x,y); add(y,x);
		}
		dfs(1,0);
		printf("%d\n",min(dp[1][2],dp[1][0]));
		scanf("%d",&n); if(n==-1) break;
	}
	return 0;
}