Codeforces Round #668 (Div. 2) D - Tree Tag（使用fa代替vis[]的dfs求树的直径+dfs求ab的距离）

Codeforces Round #668 (Div. 2) D - Tree Tag（使用fa代替vis[]的dfs求树的直径+dfs求ab的距离）

题目

http://codeforces.com/contest/1405/problem/D

题意

a和b都在树上，a要追到b的话a胜利，否则b胜利。
a先手，并且a最多走da步，b最多走db步。

题解

a胜利有两种情况，第一种情况是a第一步直接就可以碰到b，a直接胜利。
第二种情况是db<=2da，因为a可以把b逼到死胡同（树最远的子节点）此时b需要掉头，这个掉头的距离必须大于2da，否则a下一步必定追上。
如果树的直径dis<=2da，那么da在中间可以到达任意一点，a也是必胜的。
所以当min(dis,db)<=2a时，a必胜。

代码

这个dfs用fa代替vis[]，更加节省空间时间。以后可以常用。

#include<iostream>
#include<stdio.h>
#include<string.h>
#include<math.h>
#include<algorithm>
#include<vector>
#include<map>
#include<set>
#include<queue>
#include<stack>
using namespace std;
#define ll long long
#define ld long double
#define INF 0x3f3f3f3f
#define mem(a,b) memset(a,b,sizeof(a))
#define PI acos(-1)
const int maxn = 1e5+5;
const ll mod = 1e9+9;
const double eps = 1e-6L;
vector<int> g[maxn];
int vis[maxn];
int ans,dis,index;
void dfs1(int a,int b,int fa,int step)
{
	if(ans!=0) return;
	if(a == b)
	{
		ans = step;
		return;
	}
	if(g[a].size() == 1 && fa != -1) return;
	for(auto x: g[a])
	{
		if(x != fa)
		{
			dfs1(x,b,a, step+1);
		}
	}
}
void dfs2(int a,int fa,int step)
{
	if(g[a].size() == 1 && fa != -1) 
	{
		if(dis<step)
		{
			dis = step;
			index = a;
		}
	}
	for(auto x: g[a])
	{
		if(x != fa)
		{
			dfs2(x,a, step+1);
		}
	}
}
int main()
{
	int T=1;
	scanf("%d",&T);
	while(T--)
	{
		int n,a,b,da,db;
		scanf("%d %d %d %d %d",&n,&a,&b,&da,&db);
		for(int i=1; i<=n; i++)	g[i].clear(),vis[i]=0;
		for(int i=0; i<n-1; i++)
		{
			int u,v;
			scanf("%d %d",&u,&v);
			g[u].push_back(v);
			g[v].push_back(u);
		}
		ans=0;
		dfs1(a,b,-1,0);
		dis=0,index=0;
		dfs2(a,-1,0);
		dfs2(index,-1,0);
		bool flag = false;
		if(ans<=da)	flag = true;
		else if(min(db,dis)<=2*da)	flag = true;
		if(flag)	puts("Alice");
		else	puts("Bob");
	}
	return 0;
}