CodeForces - 813C The Tag Game (树的dfs遍历)

【传送门】http://codeforces.com/problemset/problem/813/C

【题目大意】两个人玩游戏，一个人跑一个人追，轮流决策，可以走也可以不走。给你一棵树，想要从某个结点到达另一个结点当且仅当两个结点有边相连，A从根节点1号点出发，B从X号点出发，B要尽可能躲A，A要尽可能抓住A，最后两个人共决策了多少次。

【题解】可以知道这样一个事实，B一定会躲到离他最远的那个结点，然后A不管怎样决策B都躲在那里不动了。但是有一点需要注意，B需要比A先到达那个叶子结点。

所以问题转化成了分别求A，B到各个叶子结点的距离。选取一个A到达某一叶子结点的最远值，注意还必须满足这个最远值要大于B到达这个叶子结点的距离。

使用DFS对树进行搜索，直到叶子结点，每每遇见一个叶子结点就记下距离。起点分别设置为1和X，距离保存在两个数组中。

【AC代码】

#include<iostream>
#include<cstdio>
#include<string>
#include<algorithm>
#include<vector>
#include<cstring>
#include<iomanip>
using namespace std;
typedef long long ll;
const int maxn = 2e5+10;
vector<int> G[maxn];
ll d1[maxn];
ll d2[maxn];

void dfs(int st, int fr, ll step, ll d[]){
    if(G[st].size() == 1 && st != 1){
        d[st] = step;
        //return ;//这里return会错误 
    }
    
    for(int i=0; i<G[st].size(); i++){
        if(G[st][i] != fr){
            dfs(G[st][i] , st, step+1, d);
        }
    }
}

int main(){
    int n,x;
    int s,t;
    while(scanf("%d%d",&n,&x) != EOF){
        ll ans = 0;
        for(int i=0; i<maxn; i++)    G[i].clear(); 
        memset(d1 , 0 ,sizeof d1);
        memset(d2 , 0 ,sizeof d2);
        while(scanf("%d%d",&s,&t) != EOF){
            G[s].push_back(t);
            G[t].push_back(s);
        }
        dfs(1, -1, 0, d1);
        dfs(x, -1, 0, d2);
        
        for(int i=2; i<=n; i++){
            if(d1[i] > d2[i] && G[i].size() == 1){
                ans = max(ans , 2*d1[i]);
            }
        }
        printf("%lld\n",ans);
    }
    return 0;
}