二叉树的编号--小球下落

 

https://vjudge.net/problem/UVA-679

有一颗二叉树，最大深度为D，且所有叶子的深度都相同。所有结点从上到下从左到右编号为1,2,3，....，2^D-1.在结点1处放一个小球，它会往下落。每个内点上都有一个开关，初始全部关闭，当每次有小球落到一个开关上时，状态都会改变。当小球到达一个内结点时，如果该结点上的开关关闭，则往左走，否则往右走。直到走到叶子结点。

一些小球从结点1处依次下落，最后一个小球将会落到哪里呢？输入叶子深度D和小球个数I，输出第I个小球最后所在的叶子编号，假设I不超过整棵树的叶子个数。D<=20；

样例输入：

4  2

3  4

10  1

2  2

8  128

16 12345

样例输出：

12

7

512

3

255

36358

分析：二叉树有个重要的性质是其k的左儿子、右儿子的编号为2k,2k+1;

为了简洁代码，可以直接把开关这种名词用0 1代表关和开。

下面代码中的：1<<maxn代表着2^maxn.

由于2^maxn太大，数组可能会出界导致running error,优化后的代码下次给出

代码：

#include<iostream>
#include<string.h>
using namespace std;
const int maxn = 20;
int a[1 << maxn];//结点的最大数为2^maxn-1个
int main()
{
	int D, I;
	while (cin >> D >> I)
	{
		memset(a, 0, sizeof(a));
		int k;
		int n = (1 << D) - 1;
		for (int i = 0; i < I; i++)//I个球下落；
		{
			k = 1;//每个球都是从顶点开始下落的
			for (; ; )
			{
				a[k] = !a[k];
				k = a[k] ? 2 * k : 2 * k + 1;
				if (k > n) break;//当k>n时，说明球已经超出叶子结点标号的，代表着已经出界了
			}
		}
		cout << k / 2 << endl;//出界前的上一个结点，即左右儿子k的父节点是k/2
	}
	return 0;
}