7-12 金字塔游戏分数 25

幼儿园里真热闹，老师带着孩子们做一个名叫金字塔的游戏，游戏规则如下：

首先，老师把孩子们按身高从高到矮排列，选出最高的做队长，当金字塔的塔顶，之后在其余小朋友里选出两个最高的，站在队长的前面，队长左右手分别牵着他们的衣服，稍高的在左边，稍矮的在右边。新加入的小朋友依次作为新的队长，继续在其余小朋友里选出高的，按照相同的规则，新队长左右手分别牵着他们的衣服，依次加入到金字塔中。这样下来之后，全幼儿园的小朋友都牵在一起，组成一个大金字塔了。 问题来了，老师想知道两个小朋友之间间隔有多少个小朋友，你能帮帮老师吗？

输入格式:

首先输入小朋友的数量N（不超过1000），之后N行，分别输入小朋友的姓名和身高，中间用空格隔开，姓名由不超过20个字符的字母组成，身高是50到130之间的整数。之后输入查询的数量Q（不超过100），随后是Q行查询，每行输入两个小朋友的姓名，中间用空格隔开。

输出格式:

请对每一个查询，在一行中输出两个小朋友之间间隔的小朋友的数量，如果输入的任意一个姓名找不到，则输出Name not found!，我们假定身高相同的小朋友，按姓名的字典序来确定顺序，字典序小的排前面，题目保证没有身高相同且姓名相同的小朋友。

输入样例:

在这里给出一组输入。例如：

5
Sandro 120
Dracon 110
Adela 120
Solmyr 100
Luna 90
3
Dracon Sandro
Dracon Luna
Luna Kyrre

输出样例:

在这里给出相应的输出。例如：

1
2
Name not found!

代码长度限制

16 KB

时间限制

100 ms

内存限制

64 MB

栈限制

8192 KB

#include<bits/stdc++.h>
using namespace std;
#define int long long
#define pb push_back
const int N = 1e6+10;
int n,m,k,T;
struct node{
	string name;
	int l,r,v;
}s[N];
map<string,string>fat;
map<string,int>dep;
bool cmp(node a , node b){
	if(a.v!=b.v)return a.v>b.v;
	return a.name<b.name;
}
void dfs(int u,string fa){
	if(u>n)return ;
	int l = u << 1 ;
	int r = u << 1 | 1 ;
	dep[s[l].name]=dep[s[u].name]+1;
	fat[s[l].name]=s[u].name;
	dfs(l,s[u].name);
	dep[s[r].name]=dep[s[u].name]+1;
	fat[s[r].name]=s[u].name;
	dfs(r,s[u].name);
}
int get(string a , string b){
	if(dep[a]<dep[b]){
		swap(a,b);
	}
	int sum = 0 ;
	while(a!=b){
		if(dep[a]>dep[b]){
			a=fat[a];
			sum++;
		}else{
			a=fat[a];
			b=fat[b];
			sum+=2;
		}
	}
	return sum;
}
signed main(){
	cin >> n;
	for(int i=1;i<=n;i++){
		cin >> s[i].name >> s[i].v;
	}
	sort(s+1,s+1+n,cmp);
	dep[s[1].name]=1;
	dfs(1,"");
	cin >> m ;
	string a , b ;
	while(m--){
		cin >> a >> b ;
		if(!dep.count(a)||!dep.count(b)){
			cout << "Name not found!" << endl;
			continue;
		}
		if(a==b){
			cout << 0 << endl;
			continue;
		}
		cout << get(a,b)-1 << endl;
	}
	return 0;
}

//7
//Adela 120
//Sandro 120
//Dracon 110
//Solmyr 100
//Luna 90
//Tazar 90
//Hack 88