7-7 金字塔游戏 (25 分)

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!

基本思路

1. 创建包含身高和姓名的结点node，并申请结点的数组p[]
2. 将n个孩子的信息保存到数组p[]中，从数组1号元素位置开始存储，0号元素不用
3. 将n个孩子按照身高由大到小和姓名字母序进行排序；排序时切记只对数组p[]中1到n号元素排序，排序后数组p[]即为一颗符合本题要求的完全二叉树；
4. 对建好的二叉树进行前序遍历，对所有结点建立以孩子姓名为键和结点序号为值的键值对映射
5. 在键值对映射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;
}