Codeup Problem A: 二叉排序树

原题地址

codeup.hustoj.com/problem.php?cid=100000611&pid=1

解题思路

计算出结点为m和n分别对应的高度h和H，H即为这个二叉树的高度。

题目中的意思是求以m为根节点的这个子树的总结点数目，那么我们要找出m子树的高度，这可以通过H-h+1计算得到。但是具体结点的数目还需要分类讨论。

经过观察可以发现，这里的父节点和孩子结点之间有关系的，假设结点为k，其左孩子为2k，右孩子为2k+1~那么我们可以通过这个关系计算出结点m对应的子树最后一层的左结点L和右结点R拉~~这最后一层也就是结点n所在的这层。

由于n的位置不确定，所以最后进行一下分类讨论，分别算出不同情况对应的结点个数就可以了~

下面是我随便写的一个图QWQ

参考代码

#include <bits/stdc++.h>
using namespace std;
#define pb push_back
typedef double db;
typedef long long LL;
typedef vector<int> VI;
const int inf = 2e9;
const LL INF = 8e18;
const int maxn = 5e5 + 5;

int main() {
	int m, n;
	while (1) {
		scanf("%d%d", &m, &n);
		if (!m && !n) break;
		int H, h, times, num;
		H = int(log(n) / log(2))+ 1;
		h = int(log(m) / log(2))+ 1;
		times = H - h;
		int left = m, right = m;
		while (times--) {
			left = left * 2;
			right = right * 2 + 1; 
		}
		if (right <= n) {
			num = pow(2, H - h + 1) - 1;
		} else if (left <= n && n < right) {
			num = pow(2, H - h) + (n - left);
		} else {
			num = pow(2, H - h) - 1;
		}
		printf("%d\n", num);
	}
	return 0;
}