比赛 20160420s 评测结果 AAAAAAAAAA
题目名称 树上的游戏 最终得分 100
用户昵称 mikumikumi 运行时间 6.080 s
代码语言 C++ 内存使用 37.70 MiB
提交时间 2016-04-20 10:17:23
显示代码纯文本
#include<cstdio>
#include<iostream>
#include<vector>
#include<cmath>
using namespace std;
const int SIZEN=200010;
int N;
pair<int,int> P[SIZEN];
vector<int> s[SIZEN];
void read()
{
	scanf("%d",&N);
	int fr,to;
	for(int i=1;i<N;i++)
	{
		scanf("%d%d",&fr,&to);
		P[i]=make_pair(fr,to);
		s[fr].push_back(to);
		s[to].push_back(fr);
	}
}
int dep[SIZEN],dfn[SIZEN*2],dfslist=0,first[SIZEN];
int ST[SIZEN*2][20];
void dfs(int x,int fa)
{
	dfn[++dfslist]=x;
	first[x]=dfslist;
	for(int i=0;i<s[x].size();i++)
	{
		int v=s[x][i];
		if(v==fa) continue;
		dep[v]=dep[x]+1;
		dfs(v,x);
		dfn[++dfslist]=x;
	}
}
void make_ST()
{
	int t=int(log2(dfslist+0.0));
	for(int i=1;i<=dfslist;i++) ST[i][0]=dfn[i];
	for(int i=1;i<=t;i++)
	{
		for(int j=1;j<=dfslist;j++)
		{
			if(j+(1<<i)-1>dfslist) break;
			if(dep[ST[j][i-1]]<dep[ST[j+(1<<(i-1))][i-1]]) ST[j][i]=ST[j][i-1];
			else ST[j][i]=ST[j+(1<<(i-1))][i-1];
		}
	}
}
void prepare()
{
	dep[1]=0;
	dfs(1,0);
	make_ST();
}
int LCA(int x,int y)
{
	int l=first[x],r=first[y];
	if(l>r) swap(l,r);
	int m=int(log2(r-l+1.0));
	int a=ST[l][m],b=ST[r-(1<<m)+1][m];
	if(dep[a]<dep[b]) return a;
	return b;
}
int dist(int x,int y)
{
	int A=LCA(x,y);
	return dep[x]+dep[y]-2*dep[A];
}
void work()
{
	prepare();
	int Q;
	scanf("%d",&Q);
	int x,y,z;
	for(int i=1;i<=Q;i++)
	{
		scanf("%d%d%d",&x,&y,&z);
		int fr=P[z].first,to=P[z].second;
		if(dep[fr]>dep[to]) swap(fr,to);
		if(dep[x]>dep[y]) swap(x,y);
		//cout<<x<<" "<<y<<" "<<fr<<" "<<to<<endl;
		int t=LCA(x,y);
		if(LCA(x,fr)==fr&&LCA(x,to)==to&&LCA(t,fr)==t&&LCA(t,to)==t) printf("NO\n");
		else if(LCA(y,fr)==fr&&LCA(y,to)==to&&LCA(t,fr)==t&&LCA(t,to)==t) printf("NO\n");
		else printf("YES\n");
	}
}
int main()
{
	freopen("treegame.in","r",stdin);
	freopen("treegame.out","w",stdout);
	read();
	work();
	return 0;
}