旅游 树形dp

题目：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;
}