欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

例题6-6 UVa679-Dropping Balls(满二叉树、递归)

程序员文章站 2022-03-14 19:19:26
...

这题解法同样很多,紫书上直接模拟最后一颗小球的做法很妙,在此就不再赘述。想记录的方法是递归解法。
列出前四层每个小球的落点,以第2和第3层为例:第二层是2,3;第三层是4,6,5,7;不难注意到,4,6分别是22与32,而5,7分别是22+1与32+1;在其他层里也可以找到这样的规律,所以可以使用以下的递归解法。

题目链接:UVa 679

AC代码:

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

int tree(int D, int I) {
	if (D == 1)
		return 1;
	else {
		int half = pow(2, D - 2);
		if (I <= half)
			return 2 * tree(D - 1, I);
		else
			return 2 * tree(D - 1, I - half) + 1;  //使I落入前half个结点
	}
}
int main() {
	int T, D, I;
	cin >> T;
	while (T--) {
		cin >> D >> I;
		cout << tree(D, I) << endl;
	}
	cin >> I;  //吞掉-1
	return 0;
}

记录紫书的解法:

#include <iostream>
using namespace std;

int main() {
	int T, D, I;
	cin >> T;
	while (T-- && cin >> D >> I) {
		int k = 1;
		for (int i = 0; i < D-1; i++) {
			if (I % 2) {
				k = k * 2;   //I是奇数,向左子树走
				I = (I + 1) / 2;  //计算I是第几个来到这个节点的小球
			}
			else {
				k = k * 2 + 1;
				I = I / 2;
			}
		}
		cout << k << endl;
	}
	return 0;
}