UVa 679 Dropping Balls 下落的小球

題目描述:

有一棵二叉樹,最大深度爲D,且所有葉子的深度都相同。所有葉子從上到下從左到右編號爲1,2,3……2^D-1。在結點1處放一個小球,它會往下落。每個內結點上都有一個開關,初始全部關閉,當每次有小球落到一個開關上時,狀態都會改變。當小球到達一個內結點時,如果該結點上的開關關閉,則往左走,否則往右走,直到走到葉子結點,如圖所示。


一些小球從結點1處依次下落,最後一個小球會落到那裏呢?輸入葉子的深度D和小球的個數I,輸出第I個小球最後所在的葉子編號。假設I不超過整棵樹的葉子個數。D<=20。輸出最多包含1000組數據。


樣例輸入:

4   2

3   4

10   1

2   2

8   128

16   12345


樣例輸出:

12

7

512

3

255

36358



思路:這是紫書上二叉樹的開篇題。作爲初學者(當然包括我)有幾點需要注意:

    1.結點k的左子結點爲2k,右爲2k+1.

    2.有2^d個結點的完全二叉樹(d爲高度),最大結點數爲    2^d-1,最大結點編號爲2^d-1.

用s數組模擬開關,每輸入一組數據初始化爲關閉一次。出界前葉子的編號爲出界時所在節點的1/2.

二叉樹遍歷。


代碼:

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>

#define loop( i, a, b ) for( int i = a; i < b; i++ )
#define fill( x ) memset( x, 0, sizeof( x ) )
#define maxd  20

using namespace std;

int n, D, T;
int s[1<<maxd];

int main() {
//	scanf( "%d", &n );
	scanf( "%d%d", &D, &T ); 
	fill( s );	//開關。
	int k, n = ( 1 << D ) - 1;	//n是最大結點編號。
	loop( i, 0, T ) {	//連續讓小球落下。
		k = 1;
		for(;;) {
			s[k] = !s[k];
			k = s[k] ? k * 2 : k * 2 + 1;	//根據開關狀態選擇下落方向。
			if( k > n )	break;	//出界。
		}
	}
	printf( "%d\n", k / 2 );
	return 0;
}

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章