万恶的remote judge
查看原帖
万恶的remote judge
250699
mot1ve楼主2020/10/3 11:22

LCA模板,第一个点就WA了,不知道是不是被卡格式了,样例可过,

#include<bits/stdc++.h>
using namespace std;
int T,idx,t,n,root,q,tot;
int f[1010][30],dep[1010],head[1010];
bool vis[1010];
struct node{
	int nxt,to;
}edge[1000010];
void add(int u,int v)
{
	edge[++idx].nxt=head[u];
	edge[idx].to=v;
	head[u]=idx;
} 
void dfs(int now,int fath)//预处理 
{
	dep[now]=dep[fath]+1;
	f[now][0]=fath;
	for(int i=1;i<=t;i++)
	{
		f[now][i]=f[f[now][i-1]][i-1];//倍增数组 
	}
	for(int i=head[now];i;i=edge[i].nxt)
	{
		int v=edge[i].to;
		dfs(v,now);
	}
}
int LCA(int x,int y)
{
	if(dep[x]<dep[y])
	swap(x,y);
	for(int i=t;i>=0;i--)//先调整到同一深度 
	{
		if(dep[f[x][i]]>=dep[y])
		{
			x=f[x][i];
			if(x==y)
			return x;
		}
	}
	for(int i=t;i>=0;i--)//一起往上跳
	{
		if(f[x][i]!=f[y][i])
		{
			x=f[x][i];
			y=f[y][i];
		}
	}
	return f[x][0]; 
}
int main()
{
	cin>>T;
	while(T--)
	{
		memset(vis,0,sizeof(vis));
		memset(head,0,sizeof(head));
		tot++,idx=0;
		scanf("%d",&n);
		t=(int)(log(n)/log(2))+1;
		for(int i=1;i<=n;i++)
		{
			int cnt;
			scanf("%d",&cnt);
			for(int j=1;j<=cnt;j++)
			{
				int v;
				scanf("%d",&v);
				add(i,v);
				vis[v]=1;//v有父亲 
			}
		}
		for(int i=1;i<=n;i++)
		{
			if(!vis[i])//找没父亲的点,就是根 
			{
				root=i;
				break;
			}
		}
		dfs(root,0);
		scanf("%d",&q);
		printf("Case %d:\n",tot);
		for(int i=1;i<=q;i++)
		{
			int x,y;
			scanf("%d%d",&x,&y);
			printf("%d\n",LCA(x,y));
		}
	}
	return 0;
} 
2020/10/3 11:22
加载中...