北京大学考研机试题：二叉树

一、题目原文

如上所示，由正整数 1，2，3……组成了一颗特殊二叉树。

我们已知这个二叉树的最后一个结点是 n。

现在的问题是，结点 m 所在的子树中一共包括多少个结点。

比如，n=12，m=3，那么上图中的结点 13,14,15 以及后面的结点都是不存在的，结点m所在子树中包括的结点有 3,6,7,12，因此结点m的所在子树中共有 4个结点。

输入格式

输入数据包括多行，每行给出一组测试数据，包括两个整数 m,n。

最后一行 0 0 表示输入结束。

输出格式

对于每一组测试数据，输出一行，该行包含一个整数，给出结点 m所在子树中包括的结点的数目。

数据范围

1≤m≤n≤10^9
最多包含 20组数据。

输入样例：

3 12
0 0

输出样例：

4

 

二、题目分析

在做题之前，读者需了解几个简单的概念性知识：

1、题目最终要求的是以m号节点为根的子树的节点个数

2、题目中的二叉树从1开始编号，那么给定根节点为m，其左孩子编号为2m，其右孩子编号为2m+1（读者应区别于从0开始编号的二叉树的左右孩子计算方法）

3、高为h的满二叉树，其节点总数为2^h−1

本题核心要点：讨论当m取不同数值时对其子树形态的影响，从而对子树的节点个数产生的影响

这样教科书式的语言可能不好理解，下面作者将给出具体示例演示以上变化。

此处，先假定存在两棵满二叉树，分别是以1号节点为根的“大树”和以m号节点为根的“小树”。

综上，我们的代码思路应当如下：

1、求“大树”和“小树”的根节点所在层数（从编号为1的节点开始算第一层，注意这里求的不是两树的树高）

2、设计访问左孩子、右孩子的算法（得出左孩子和右孩子的编号）

3、利用2所求的左右孩子编号与变量n对比，得出“小树”的形态

4、利用1所求的层数来求“小树”的节点个数

三、代码书写

#include <cstdio>
#include <iostream>
using namespace std;

int main(){
	int m,n;
	while(scanf("%d%d",&m,&n)!=EOF){ //题目要求有多组数据
		if(m==0&&n==0){  //当m=0，n=0时结束
			break;
		}

	int i=1;  //i表示预置“大树”和“小树”的层数为1
	int bigtall,smalltall;  //“大树”和“小树”的层数
	while(1){
		if(m>=1<<(i-1)&&m<=(1<<i)-1){  //求m所在层数
			smalltall=i;
		}
		if(n>=1<<(i-1)&&n<=(1<<i)-1){  //求n所在层数，即“大树”有多少层
			bigtall=i;
			break;
		}
		i++;
	}
	
	int leftchild=m;  //从编号为m的节点开始往下访问，求其左孩子、右孩子的编号
	int rightchild=m;  //此处是加上不存在的节点的高度，后面会减去，不影响题目最终结果
	for(i=smalltall;i<bigtall;i++){
		leftchild=leftchild*2;
		rightchild=rightchild*2+1;
	}
	
	int num;  //num为“小树”的节点个数
	if(n>rightchild){  //情况1，“小树”为高度hm的满二叉树
		num=(1<<(bigtall-smalltall+1))-1; 
	}else if(n>=leftchild){  //情况2，“小树”为高度hm的完全二叉树
		num=(1<<(bigtall-smalltall))-1+(n-leftchild+1);
	}else{  //情况3，“小树”为高度hm-1的满二叉树
		num=(1<<(bigtall-smalltall))-1;
	}
	
	printf("%d\n",num);
	
	}
	
	return 0;
}

 

四、OJ测试