HDU 4409 Family Name List --乱搞、LCA

时间:2023-12-29 16:42:20

题意: 给出一些名字,名字间有父子关系,有三种操作:

1.按祖先到后代,兄弟间按字典序由小到大排序,然后输出

2.求某个节点的兄弟节点有多少个,包括自己(注意,根节点的兄弟节点是1)

3.求节点a和b的公共祖先 (注意:如果公共祖先是a或b,必须要输出其父亲,与传统的LCA可以是自己不同)

解法: 先把整棵树整理出来,son[u]表示u的儿子个数,用来求兄弟个数, fa[u]表示父亲,Gson存储儿子的标号,关于排序的问题,先读入所有名字,然后排个序哈希一下,使字典序小的节点标号一定小,那么直接sort(Gson.begin(),Gson.end()) 就将儿子排序了。

然后用RMQ 搞在线LCA 即可

代码:

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cmath>
#include <algorithm>
#include <string>
#include <vector>
#include <map>
using namespace std;
#define N 30107 map<string,int> mp;
int son[N],fa[N],cnt[N];
vector<int> Gson[N];
string TS[N],NS[N];
int ati[],f[],bn,b[],dp[][],ind; void dfs(int u,int dep)
{
for(int i=;i<dep;i++) printf(".");
cout<<NS[u]<<endl;
for(int i=;i<Gson[u].size();i++)
{
int v = Gson[u][i];
dfs(v,dep+);
}
} void init()
{
memset(ati,,sizeof(ati));
memset(f,,sizeof(f));
memset(b,,sizeof(b));
memset(dp,,sizeof(dp));
bn = ind = ;
} void dfs_2(int u,int father)
{
int tmp = ++ind;
f[tmp] = u;
b[++bn] = tmp;
ati[u] = bn;
for(int i=;i<Gson[u].size();i++)
{
int v = Gson[u][i];
if(v==father) continue;
dfs_2(v,u);
b[++bn]=tmp;
}
} void RMQ_init(int n)
{
for (int i=; i<=n; i++) dp[i][]=b[i];
int m=floor(log((double)n*1.0)/log((double)2.0));
for (int j=; j<=m; j++)
for (int i=; i<=n-(<<j)+; i++)
dp[i][j]=min(dp[i][j-],dp[i+(<<(j-))][j-]);
} int RMQ(int l,int r)
{
int k=floor(log((double)r-l+)/log(2.0));
return min(dp[l][k],dp[r-(<<k)+][k]);
} int LCA(int a,int b)
{
if (ati[a] > ati[b]) swap(a,b);
return f[RMQ(ati[a],ati[b])];
} int main()
{
int n,m,i,j;
while(scanf("%d",&n)!=EOF && n)
{
mp.clear();
memset(son,,sizeof(son));
memset(fa,,sizeof(fa));
string ss;
for(i=;i<=n;i++) Gson[i].clear();
for(i=;i<=n;i++)
{
cin>>TS[i];
int len = TS[i].length();
for(j=;j<len;j++)
if(TS[i][j] != '.') break;
NS[i] = TS[i].substr(j,len-j);
}
sort(NS+,NS+n+);
for(i=;i<=n;i++)
mp[NS[i]] = i;
for(i=;i<=n;i++)
{
ss = TS[i];
int len = ss.length();
for(j=;j<len;j++)
if(ss[j] != '.') break;
string S = ss.substr(j,len-j);
cnt[j] = mp[S];
if(j != )
{
fa[mp[S]] = cnt[j-]; //最近的有j-1个'.'的名字就是有j个'.'的名字的父亲
son[cnt[j-]]++;
Gson[cnt[j-]].push_back(mp[S]);
}
else
fa[mp[S]] = ;
}
for(i=;i<=n;i++)
sort(Gson[i].begin(),Gson[i].end());
for(i=;i<=n;i++)
if(fa[i] == ) break;
int father = i;
init();
dfs_2(father,);
RMQ_init(bn);
scanf("%d",&m);
char s[];
while(m--)
{
scanf("%s",s);
if(s[] == 'L')
dfs(father,);
else if(s[] == 'b')
{
cin>>ss;
int k = mp[ss];
if(k == father) puts("");
else printf("%d\n",son[fa[k]]);
}
else
{
string S1,S2;
cin>>S1>>S2;
int ms1 = mp[S1];
int ms2 = mp[S2];
int fat = LCA(ms1,ms2);
if(fat == ms1 || fat == ms2) cout<<NS[fa[fat]]<<endl;
else cout<<NS[fat]<<endl;
}
}
}
return ;
}