hihoCoder 1069 最近公共祖先·三

最新推荐文章于 2017-10-05 16:03:03 发布

原创最新推荐文章于 2017-10-05 16:03:03 发布 · 378 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#并查集

------------图论算法------------ 专栏收录该内容

47 篇文章

订阅专栏

本文介绍了一种基于并查集的家族树查询算法实现。通过使用C++编程语言，该算法能够有效地处理大规模的家族成员关系，并能快速响应关于两个个体间家族关系的查询请求。文章详细展示了如何构建家族成员的数据结构、如何进行父子关系的建立以及如何通过深度优先搜索（DFS）来查找两个个体是否属于同一家族。

#include <bits/stdc++.h>
using namespace std;

map<string,int> nameMap;
string names[100005];

struct node{
	int pa;
	vector<int> son;
	vector<pair<int,int> > qu;
	node() : pa(-1){}
}no[100005];
int ans[100005];


int findset(int x){
    if(x != no[x].pa)
        no[x].pa = findset(no[x].pa);
    return no[x].pa;
}

void dfs(int f){
	no[f].pa = f;
    int len = no[f].son.size();
	for(int i=0 ;i<len;i++){
		int index = no[f].son[i];
		dfs(index);
		no[index].pa  = f;
	}
	int qlen = no[f].qu.size();
	for(int i=0; i<qlen; i++){
		pair<int, int> p = no[f].qu[i];
		if(no[p.first].pa != -1)
			ans[p.second] = findset(p.first);
	}
}

int main(){
	int N,M;
	scanf("%d",&N);
	for(int i=0; i<N; i++){
		string fa,son;
		cin>>fa>>son;
		if(nameMap.find(fa) == nameMap.end())
			nameMap[fa] = nameMap.size();
		if(nameMap.find(son) == nameMap.end())
			nameMap[son] = nameMap.size();
		int nf = nameMap[fa], ns = nameMap[son];
		names[nf] = fa;
		names[ns] = son;
		no[nf].son.push_back(ns);
	}
	scanf("%d",&M);
	for(int i=0; i<M; i++){
		string qu1,qu2;
		cin>>qu1>>qu2;
		int n1 = nameMap[qu1], n2 = nameMap[qu2];
		no[n1].qu.push_back(make_pair(n2,i));
		no[n2].qu.push_back(make_pair(n1,i));
	}
	dfs(1);
	for(int i = 0; i < M ; i++)
		//puts(names[ans[i]].c_str());
		cout<<names[ans[i]]<<endl;
	return 0;
}