L2-2 金字塔游戏

题目来源：天梯赛训练题

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

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

输入格式:

首先输入小朋友的数量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!“；否则则输出两个孩子对应结点的最短路径中的结点个数，分两种情况讨论:

        a. 结点a和b在同一层但a!=b（若a==b则c1=0），不断将a和b同时向其父节点收敛（a=a/2;b=b/2;），直到a==b，并统计此过程中经过的结点个数c1；

        b. 结点a和b不在同一层，将b结点向上收敛（b=b/2;），直到a和b处于同一层，并统计此过程中经过的结点个数c2，然后按照情况a进行处理；

        c. 最后输出c1+c2；

        d. 计算结点所在层数使用log(int a)函数即可;

6. 在测试代码是注意测试4种情况：

        a. 至少有一名孩子结点不存在；

        b. 两个孩子所在的结点有共同的祖先结点；

        c. 两个孩子结点中一个结点为另外一个结点的祖先结点；

        d. 两个孩子结点中一个结点为另外一个结点的父结点；

       

//L2-2 金字塔游戏
#include <iostream>
using namespace std;
#include <algorithm>
#include <map>
#include <cmath>
#define N 1005

struct node
{
	string name;
	int high;
}p[N]; 
int n;

bool cmp(node x,node y)
{
	if(x.high==y.high)
		return x.name<y.name;
	return x.high>y.high;
}

map<string,int> mp; 
void preOrder(int r)		//先序遍历 
{
	if(r<=n)
	{
		mp[p[r].name]=r;
		preOrder( r*2);
		preOrder( r*2+1);
	}
 } 
 
int log(int a)		// log2 a
{
	int c=0;
	while(a)
	{
		a/=2;
		c++;
	}
	return c-1;
}

int main()
{
	int i;
	cin>>n;cin.ignore();
	for(i=1;i<=n;i++)		//输入 
	{
		cin>>p[i].name>>p[i].high;cin.ignore();
	}
	
	sort(p+1,p+n+1,cmp); 		//按身高、姓名排序 ，排后的序列即为二叉树的层序遍历序列 
	
	preOrder(1); 		//先序遍历二叉树，用于建立以每个孩子姓名为键的映射
	 
	int q;
	cin>>q;
	string name1,name2;
	while(q--)
	{
		cin>>name1>>name2;			//输入待查寻的两个孩子的姓名 
		if(mp.find(name1)==mp.end()||mp.find(name2)==mp.end())	//有一个结点找不到
			cout<<"Name not found!"<<endl;
		else
		{
			int a=mp[name1];	//待查寻孩子1对应的结点编号 
			int b=mp[name2];	//待查寻孩子2对应的结点编号 
			
			if(a>b)
				swap(a,b);
			int c=0;
			while(log(a)!=log(b))		//将b向上等价到a的那一层 
			{
				b=b/2;c++;			//将b置为其父节点 
			}
			if(a==b)
			{
				if(c==0)			//a和b原本就相同 
					cout<<0<<endl;
				else				//b等价后与a相同 
					cout<<c-1<<endl;
			}
			else
			{
				while(a!=b)		//a和b同时向上等价 
				{
					a=a/2;
					b=b/2;
					c+=2;
				}
				cout<<c-1<<endl;
			}
		}
	}
	return 0;
 }