hihocoder #1067 : 最近公共祖先·二
描述
上上回说到,小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的询问,按照在输入中出现的顺序,各输出一行,表示查询的结果:他们的所有共同祖先中辈分最低的一个人的名字。
- 样例输入
-
4 Adam Sam Sam Joey Sam Micheal Adam Kevin 3 Sam Sam Adam Sam Micheal Kevin
- 样例输出
-
Sam Adam Adam
分析:Tarjan算法,DFS+并查集,从根节点往下搜时,同时看询问里与当前点相关的点是否访问过,如果访问了,那么答案就是该访问过的点当前最远的祖先,DFS回溯和并查集维护好当前点的最远祖先就行了。
代码:#include <iostream> #include <cstdio> #include <cstdlib> #include <cmath> #include <algorithm> #include <climits> #include <cstring> #include <string> #include <set> #include <map> #include <queue> #include <stack> #include <vector> #include <list> #define rep(i,m,n) for(i=m;i<=n;i++) #define rsp(it,s) for(set<int>::iterator it=s.begin();it!=s.end();it++) #define vi vector<int> #define pii pair<int,int> #define mod 1000000007 #define inf 0x3f3f3f3f #define pb push_back #define mp make_pair #define fi first #define se second #define ll long long #define pi acos(-1.0) const int maxn=1e5+10; const int dis[4][2]={{0,1},{-1,0},{0,-1},{1,0}}; using namespace std; ll gcd(ll p,ll q){return q==0?p:gcd(q,p%q);} ll qpow(ll p,ll q){ll f=1;while(q){if(q&1)f=f*p;p=p*p;q>>=1;}return f;} int n,m,cnt,p[maxn],vis[maxn],res1[maxn],res2[maxn]; string a,b; map<int,string>name; map<string,int>id; map<pair<int,int>,string>ans; vi path[maxn],query[maxn]; void init() { for(int i=1;i<=n;i++)p[i]=i; } int all(int x) { return p[x]==x?x:p[x]=all(p[x]); } void dfs(int x) { vis[x]=1; for(int y:query[x]) { if(vis[y]) { ans[mp(x,y)]=ans[mp(y,x)]=name[all(y)]; } } for(int y:path[x]) { dfs(y); p[y]=x; } } int main() { int i,j,k,t; scanf("%d",&n); init(); while(n--) { cin>>a>>b; if(!id[a])id[a]=++cnt,name[cnt]=a; if(!id[b])id[b]=++cnt,name[cnt]=b; path[id[a]].pb(id[b]); } scanf("%d",&m); rep(i,0,m-1) { cin>>a>>b; res1[i]=id[a],res2[i]=id[b]; query[id[a]].pb(id[b]),query[id[b]].pb(id[a]); } dfs(1); rep(i,0,m-1)cout<<ans[mp(res1[i],res2[i])]<<endl; //system ("pause"); return 0; }