7-7 金字塔游戏 (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!
基本思路
- 创建包含身高和姓名的结点node,并申请结点的数组p[]
- 将n个孩子的信息保存到数组p[]中,从数组1号元素位置开始存储,0号元素不用
- 将n个孩子按照身高由大到小和姓名字母序进行排序;排序时切记只对数组p[]中1到n号元素排序,排序后数组p[]即为一颗符合本题要求的完全二叉树;
- 对建好的二叉树进行前序遍历,对所有结点建立以孩子姓名为键和结点序号为值的键值对映射
- 在键值对映射mp中查询两个待查寻孩子的结点序号,记为a和b(其中规定a<=b),若由一个孩子的姓名不存在则输出”Name not found!“;否则则输出两个孩子对应结点的最短路径中的结点个数。
代码
#include<bits/stdc++.h>
using namespace std;
int n, m;
struct node{
string name;
int height;
};
node p[1005];
bool cmp1(node a,node b){
if(a.height!=b.height){
return a.height > b.height;
}else if(a.name!=b.name){
return a.name < b.name;
}
}
map<string,int> mp;//建立姓名和数组下标之间的映射关系
void dfs(int x){
if(x>n){
return;
}
mp[p[x].name]=x;
dfs(2*x);
dfs(2*x+1);
}
//可以得到编号为a的结点所在的层次(根结点为第0层,向下依次递增)
int log(int a){
int c=0;
while(a){
a/=2;
c++;
}
return c-1;
}
int main(){
//读入所有孩子的姓名和身高
cin >> n;
for (int i = 1; i <= n;i++){
cin >> p[i].name >> p[i].height;
}
//对所有孩子按照身高降序,字典序的顺序来排序
sort(p + 1, p + n + 1, cmp1);
//先序遍历完全二叉树,建立姓名和数组下标之间的映射关系
dfs(1);
//读入测试样例并输出
cin>>m;
while(m--){
string x,y;
cin>>x>>y;
if(mp.find(x)==mp.end()||mp.find(y)==mp.end()){
cout<<"Name not found!"<<endl;
}else{
int a=mp[x];
int b=mp[y];
int c=0;//抬升的阶数
if(a>b){
swap(a,b);
}else if(a==b){
cout<<0<<endl;
continue;
}
//将b向上等价到a的那一层
while(log(a)!=log(b)){
b=b/2;//将b置为其父节点
c++;
}
//如果a==b,说明a与b抬升到同一层次后是同一个结点,即a,b在同一条路径上,输出(抬升的阶数-1)==(两个结点路径之间的结点数)
//否则,a和b同时向上等价,直到a和b相等,输出(抬升的阶数-1)==(两个结点路径之间的结点数)
if(a==b){
cout<<c-1<<endl;
}else{
while(a!=b){
a=a/2;
b=b/2;
c=c+2;
}
cout<<c-1<<endl;
}
}
}
return 0;
}