COCI 2017/2018 Round #2,November 4th,2017 K Usmjeri [LCA+并查集]

题意:给一个树,有m对点(u,v),将树上的每条无向边变为有向边,问有多少种情况满足m对点u能到达v或u能到达v。

题解:对于u与v,我们找到他们的LCA,对于从u->lca(u,v)的路径边的方向是一致的,v->lca(u,v)也是。我们新开n个节点n+1~2*n,表示边i+n与i的方向不同,于是对于m个询问,我们将u->lca,v->lca的边并查集连接到一起,他们的相反边也同样,然后将u->lca的边与v+n->lca+n的边并在一起,v->lca与u+n->lca+n的边并在一起,表示一条询问路径。若i与i+n的边在同一个集合,就表示不可能有这样的方案,否则,答案是2^(并查集数量/2)。

AC代码:

#include
#include
#include
#include
#define N 300005
#define mod 1000000007
using namespace std;
vectorvt[N];
int fa[N],deep[N];
int p[N][20];
int pre[N*2];
int n,m;
void dfs(int u,int FA)
{ 
	fa[u]=FA;
	deep[u]=deep[FA]+1;
	p[u][0]=FA;
    for(int i=0;i=0;j--) //使a,b两点的深度相同 
        if(deep[a]-(1<=deep[b]) a=p[a][j]; 
    if(a==b)return a; 
    for(j=i;j>=0;j--) { 
        if(p[a][j]!=-1&&p[a][j]!=p[b][j]) { 
            a=p[a][j]; b=p[b][j]; 
        }
    } 
    return p[a][0]; 
}
int find(int i)
{
	if(i!=pre[i])return pre[i]=find(pre[i]);
	return pre[i];
}
void merge(int u,int LCA)
{
	while(deep[p[u][0]]>deep[LCA])
	{
		int f=p[u][0];
		pre[find(u)]=find(f);
		pre[find(u+n)]=find(f+n);
		u=find(f);
	}
}
void merge2(int u,int v)
{
	pre[find(u+n)]=find(v);
	pre[find(v+n)]=find(u);
}
__int64 qmi(__int64 a,__int64 b)
{
	__int64 ans=1;
	while(b)
	{
		if(b%2)ans=ans*a%mod;
		a=(a*a)%mod;
		b/=2;
	}
	return ans;
}
int a[N],b[N],LCA[N];
int main()
{
	int i,j;
	scanf("%d%d",&n,&m);
	memset(p,-1,sizeof(p));
	for(i=1;i<=2*n;i++)
		pre[i]=i;
	for(i=0;i


你可能感兴趣的:(LCA,并查集)