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<stdio.h>
#include<vector>
#include<string.h>
#include<algorithm>
#define N 300005
#define mod 1000000007
using namespace std;
vector<int>vt[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<vt[u].size();i++)
    {
    	int to=vt[u][i];
    	if(FA==to)continue;
 		dfs(to,u);
    }
}
void init() 
{ 
    int i,j;
    for(j=1;(1<<j)<=n;j++) 
		for(i=1;i<=n;i++) 
			if(p[i][j-1]!=-1) p[i][j]=p[p[i][j-1]][j-1];
}

int lca(int a,int b) {//最近公共祖先 
    int i,j; 
    if(deep[a]<deep[b])swap(a,b); 
    for(i=0;(1<<i)<=deep[a];i++); i--; 
    for(j=i;j>=0;j--) //使a，b两点的深度相同 
        if(deep[a]-(1<<j)>=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<n-1;i++)
	{
		int u,v;
		scanf("%d%d",&u,&v);
		vt[u].push_back(v);
		vt[v].push_back(u);
	}
	deep[1]=1;
	dfs(1,1);
	init();
	for(i=0;i<m;i++)
	{
		scanf("%d%d",&a[i],&b[i]);
		LCA[i]=lca(a[i],b[i]);
		merge(a[i],LCA[i]);
		merge(b[i],LCA[i]);
	}
	for(i=0;i<m;i++)
		if(LCA[i]!=a[i]&&LCA[i]!=b[i])
			merge2(a[i],b[i]);
	for(i=1;i<=n;i++)
	{
		int t1=find(i);
		int t2=find(i+n);
		if(t1==t2)
		{
			printf("0\n");
			return 0;
		}
	}
	int sum=0;
	for(i=2;i<=n;i++)
		sum+=(pre[i]==i);
	for(i=n+2;i<=2*n;i++)
		sum+=(pre[i]==i);
	printf("%I64d\n",qmi(2,1*sum/2));
	return 0;
}