题目:https://ac.nowcoder.com/acm/problem/15748
Cwbc和XHRlyb生活在s市,这天他们打算一起出去旅游。
旅行地图上有n个城市,它们之间通过n-1条道路联通。
Cwbc和XHRlyb第一天会在s市住宿,并游览与它距离不超过1的所有城市,之后的每天会选择一个城市住宿,然后游览与它距离不超过1的所有城市。
他们不想住在一个已经浏览过的城市,又想尽可能多的延长旅行时间。
XHRlyb想知道她与Cwbc最多能度过多少天的时光呢?
聪明的你在仔细阅读题目后,一定可以顺利的解决这个问题!
输入描述:
第一行,两个正整数n和s,表示城市个数和第一天住宿的城市s。
接下来n-1行,每行两个整数x和y,表示城市x与城市y之间有一条双向道路。
输出描述:
第一行,一个非负整数表示答案。
示例1
输入
复制
4 1
1 2
2 3
3 4
输出
复制
2
说明
第一天,在1号城市住宿,游览了1、2号城市。
第二天,在3号城市住宿,游览了4号城市,旅行结束。
备注:
1 ≤ n ≤ 500000, 1 ≤ s, x, y ≤ n。
题解,这是一个树形dp入门题。
要使旅游的天数最多,用dp即可,因为只要住了必然有一天,所以所有的dp[u][1]=1。dp[u][0]表示不住u城市能待的最长天数,dp[u][1]表示住在u城市能待的最长时间。那么因为u住了之后相邻的节点v必然为0,所以有dp[u][1]+=dp[v][0],如果u不住,那么相邻节点v可住可不住,那就选取最大值。就有dp[u][0]+=max(dp[v][1],dp[v][0])
同时用dfs进行dp的时候,由于存图的时候是一个双向图,传入参数为u,fa,如果等于父节点那么就跳过。
代码:
#include <bits/stdc++.h>
#include <algorithm>
#include<iostream>
#include <stdio.h>
const int maxn=5e5+5;
using namespace std;
typedef long long ll;
vector<int>g[maxn];
int r[maxn];
int dp[maxn][2];//该节点选不选的最长时间
//int ans=0;
void dfs(int u,int fa){
dp[u][1]=1;//只要选了必有一天
for(int i=0;i<g[u].size();i++){
int v=g[u][i];
if(fa==v)
continue;
dfs(v,u);
dp[u][1]+=dp[v][0];
dp[u][0]+=max(dp[v][1],dp[v][0]);
}
}
int main (){
int n,s,x,y;
cin>>n>>s;
for(int i=1;i<n;i++){
cin>>x>>y;
g[x].push_back(y);
g[y].push_back(x);
}
dfs(s,0);
cout<<dp[s][1]<<endl;
return 0;
}