旅游（树形dp）

 

题目描述

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

4 1
1 2
2 3
3 4

输出

复制 2

2

说明

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

备注:

1 ≤ n ≤ 500000, 1 ≤ s, x, y ≤ n。

#include<bits/stdc++.h>
using namespace std;
typedef pair<int,int> PII;
int n,s,cnt;
vector<int> g[500005];

int dp[500005][2];

void dfs(int x,int father)
{
    for(int i=0;i<g[i].size();i++)
    {
        if(g[x][i]==father) continue;
        dfs(g[x][i],x);
        dp[x][0]+=max(dp[g[x][i]][0],dp[g[x][i]][1]);
        dp[x][1]+=dp[g[x][i]][0];
    }
    dp[x][1]++;
}

int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    cin>>n>>s;
    cnt=0;
    int x,y;
    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;
}