#1067 : 最近公共祖先·二
-
4 Adam Sam Sam Joey Sam Micheal Adam Kevin 3 Sam Sam Adam Sam Micheal Kevin
样例输出
-
Sam Adam Adam
-
#include <iostream> #include <cstdio> #include <vector> #include <map> #include <string.h> #include <cstdlib> using namespace std; const int AX = 1e5 + 10; map<string,int>mp1; //将字符串转化为数字 map<int,string>mp2; //保存数字对应的字符串 std::vector<int>vec[AX]; std::vector<int>que[AX]; //查询记录 int n; map<pair<int,int>,int>ans; //保存查询结果 int l[AX],r[AX]; //记录查询顺序 int vis[AX],pre[AX]; //标记访问过 与 祖先节点 int find(int x){ return pre[x]==x?x:pre[x]=find(pre[x]); } void Init(){ for(int i=0;i<=n;i++){que[i].clear();vec[i].clear();vis[i]=0;pre[i]=i;}mp1.clear();mp2.clear();ans.clear(); } void Targan(int u){ for(int i=0;i<vec[u].size();i++){ Targan(vec[u][i]); //访问子树 pre[vec[u][i]] = u; //子树节点与根节点合并 } vis[u]=1; //标记此节点访问过 for(int i=0;i<que[u].size();++i){ int v=que[u][i]; if(vis[v]){ //如果访问过,那么这个节点的祖先就是其与u的共同祖先 ans[make_pair(u,v)]=find(v); //将祖先存在ans数组中 } } } int main(){ ios_base::sync_with_stdio(false); cin.tie(0); int m; while(cin>>n){ string fa,son; Init(); //初始化 for(int i=1;i<=n;i++){ //输入并转化为数字 cin>>fa>>son; if(!mp1[fa]){ mp1[fa]=++cnt; mp2[cnt]=fa; }if(!mp1[son]){ mp1[son]=++cnt; mp2[cnt]=son; } vec[mp1[fa]].push_back(mp1[son]); } cin>>m; for(int i=0;i<m;i++){ cin>>fa>>son; l[i]=mp1[fa]; r[i]=mp1[son]; que[l[i]].push_back(r[i]); que[r[i]].push_back(l[i]); } Targan(1); for(int i=0;i<m;i++){ int x; x = (ans[make_pair(l[i],r[i])] == 0?ans[make_pair(r[i],l[i])]:ans[make_pair(l[i],r[i])]); cout<<mp2[x]<<endl; } } return 0; }
描述
上上回说到,小Hi和小Ho用非常拙劣——或者说粗糙的手段山寨出了一个神奇的网站,这个网站可以计算出某两个人的所有共同祖先中辈分最低的一个是谁。远在美国的他们利用了一些奇妙的技术获得了国内许多人的相关信息,并且搭建了一个小小的网站来应付来自四面八方的请求。
但正如我们所能想象到的……这样一个简单的算法并不能支撑住非常大的访问量,所以摆在小Hi和小Ho面前的无非两种选择:
其一是购买更为昂贵的服务器,通过提高计算机性能的方式来满足需求——但小Hi和小Ho并没有那么多的钱;其二则是改进他们的算法,通过提高计算机性能的利用率来满足需求——这个主意似乎听起来更加靠谱。
于是为了他们第一个在线产品的顺利运作,小Hi决定对小Ho进行紧急训练——好好的修改一番他们的算法。
而为了更好的向小Ho讲述这个问题,小Hi将这个问题抽象成了这个样子:假设现小Ho现在知道了N对父子关系——父亲和儿子的名字,并且这N对父子关系中涉及的所有人都拥有一个共同的祖先(这个祖先出现在这N对父子关系中),他需要对于小Hi的若干次提问——每次提问为两个人的名字(这两个人的名字在之前的父子关系中出现过),告诉小Hi这两个人的所有共同祖先中辈分最低的一个是谁?
输入
每个测试点(输入文件)有且仅有一组测试数据。
每组测试数据的第1行为一个整数N,意义如前文所述。
每组测试数据的第2~N+1行,每行分别描述一对父子关系,其中第i+1行为两个由大小写字母组成的字符串Father_i, Son_i,分别表示父亲的名字和儿子的名字。
每组测试数据的第N+2行为一个整数M,表示小Hi总共询问的次数。
每组测试数据的第N+3~N+M+2行,每行分别描述一个询问,其中第N+i+2行为两个由大小写字母组成的字符串Name1_i, Name2_i,分别表示小Hi询问中的两个名字。
对于100%的数据,满足N<=10^5,M<=10^5, 且数据中所有涉及的人物中不存在两个名字相同的人(即姓名唯一的确定了一个人),所有询问中出现过的名字均在之前所描述的N对父子关系中出现过,第一个出现的名字所确定的人是其他所有人的公共祖先。
输出
对于每组测试数据,对于每个小Hi的询问,按照在输入中出现的顺序,各输出一行,表示查询的结果:他们的所有共同祖先中辈分最低的一个人的名字。