題目描述:
有一棵二叉樹,最大深度爲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;
}