UVa-679 Dropping Balls(二叉树的编号)

题目描述如下:

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

一些小球从结点1处依次开始下落,最后一个小球将会落到哪里呢?输入叶子深度D和 小球个数I,输出第I个小球最后所在的叶子编号。假设I不超过整棵树的叶子个数。D≤20。 输入最多包含1000组数据。
样例输入:

5
4 2
3 4
10 1
2 2
8 128
-1
样例输出:
12
7
512
3
255
 

有个结论非常重要,对于一个结点k,其左子结点和右子结点的编号分别为2k和2k+1。

思路一(超时):遍历每一个小球下落的情况,用一个布尔数组模拟开关,代码如下:

#include <bits/stdc++.h>
using namespace std;
#define fast                                ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
#define ll                                  long long
#define _for(i,a,b)                         for(int i = a;i < b;i++)
#define rep(i,a,b)                          for(int i = a;i <= b;i++)
#define all(s)                              s.begin(), s.end()

int kase, D, I;
bool kg[1 << 20];

int main()
{
	while(scanf("%d", &kase)!=EOF&&kase>=0)
	while (kase--)
	{
		scanf("%d%d", &D, &I);
		memset(kg, 0, sizeof(kg));
		int en = (1 << D) - 1;//最后一个结点
		int pos;
		_for(i, 0, I)
		{
			pos = 1;//初始位置
			for(;;)
			{
				kg[pos] = !kg[pos];
				pos = (kg[pos]) ? pos * 2 : pos * 2 + 1;
				if (pos > en)break;
			}
		}
		printf("%d
", pos/2);
	}
	return 0;
}

超时原因:I最大为叶子个数,也就是2^19-1个,而D最大为20,每组测试数据时间复杂度最大为(2^19-1)*20,并且输入可能有1000组数据。

思路二(AC):第a个小球只需对a奇偶性进行判断,然后每掉落一层更新一次a,判断奇偶性,直到到达叶子结点。

以题目中的i为例,当i是奇数时,它是往左走的第(i+1)/2个球,当i是偶数时,它是往右走的第i/2个球。我们只需每层更新i即可。

#include <bits/stdc++.h>
using namespace std;
#define fast                                ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
#define ll                                  long long
#define _for(i,a,b)                         for(int i = a;i < b;i++)
#define rep(i,a,b)                          for(int i = a;i <= b;i++)
#define all(s)                              s.begin(), s.end()

int kase,D, I;

int main()
{
	scanf("%d", &kase);
	while (1) 
	{
		scanf("%d", &D);
		if (D == -1) break;
		scanf("%d", &I);
		int k = 1;
		for (int i = 0; i < D - 1; i++) {//直接判断最后一个
			if (I % 2) {
				k = k * 2;
				I = (I + 1) / 2;
			}
			else {
				k = k * 2 + 1;
				I /= 2;
			}
		}
		printf("%d
", k);
	}

	return 0;
}
原文地址:https://www.cnblogs.com/theory/p/11884334.html