牛客小白月赛3 G

旅游

题目链接
考树形dp

题目描述

链接：https://ac.nowcoder.com/acm/contest/87/G
来源：牛客网

Cwbc和XHRlyb生活在s市，这天他们打算一起出去旅游。
旅行地图上有n个城市，它们之间通过n-1条道路联通。
Cwbc和XHRlyb第一天会在s市住宿，并游览与它距离不超过1的所有城市，之后的每天会选择一个城市住宿，然后游览与它距离不超过1的所有城市。
他们不想住在一个已经浏览过的城市，又想尽可能多的延长旅行时间。
XHRlyb想知道她与Cwbc最多能度过多少天的时光呢？
聪明的你在仔细阅读题目后，一定可以顺利的解决这个问题！

输入描述:

第一行，两个正整数n和s，表示城市个数和第一天住宿的城市s。
接下来n-1行，每行两个整数x和y，表示城市x与城市y之间有一条双向道路。

输出描述

第一行，一个非负整数表示答案。

输入样式

4 1
1 2
2 3
3 4

输出样式

2

说明

第一天，在1号城市住宿，游览了1、2号城市。
第二天，在3号城市住宿，游览了4号城市，旅行结束。

算法分析

树形dp,算天数f[i][1]=1;f的初始值为1天
f[i][0]=f[j][1];//f[i][0]代表 i 没来,
f[i][1]=max(f[j][0],f[j][1]);//f[i][1]代表 i 来了的情况下,判断是没来天数多还是来了的天数多

代码实现

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<math.h>
using namespace std;
typedef long long ll;
const int maxn=5e5+5; 
struct edge
{
	int pre,to;
}e[maxn*2];
int head[maxn],f[maxn][2];
int cnt=0;
void add(int x,int y)
{
	e[++cnt].pre=head[x];
	e[cnt].to=y;
	head[x]=cnt;
}
void dfs(int x,int y)
{
	for(int i=head[x];i;i=e[i].pre)
	{
		int s=e[i].to;
		if(s==y)	continue;
		dfs(s,x);
		f[x][0]+=max(f[s][1],f[s][0]);
		f[x][1]+=f[s][0];
	}
}
int main()
{
	int n,z;
	cin>>n>>z;
	for(int i=1;i<=n;i++)	
		f[i][1]=1;
	int a,b;
	for(int i=1;i<n;i++)
	{
		cin>>a>>b;
		add(a,b);
		add(b,a);
	}	
	dfs(z,0);
	cout<<f[z][1]<<endl;
	return 0;
}