从零开始的紫书刷题3：树与二叉树篇之小球下落

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

     

       输入叶子深度D和小球个数I，输出第I个小球最后所在的叶子编号。假设I不超过整棵树的叶子个数。D<=20。输入最多包含100组数据。

#include <bits/stdc++.h>
#include <iostream>
using namespace std;

const int N = 20;
//1<<N表示2的N次方；
int s[1 << N];

int main() {
	int D, I;//D为下落深度；
	while (scanf("%d%d", &D, &I) == 2) {
		//初始化数组，0表示节点处于关闭状态；
		memset(s, 0, sizeof s);
		//n表示完全二叉树最大节点编号，为2^D-1，用位运算表示(1<<D)-1；
		int k, n = (1 << D) - 1;
		for (int i = 0; i < I; i++) {
			k = 1;//k表示每一次小球所在的节点；
			for (;;) {
				s[k] = !s[k];//每次下落开关状态改变；
				k = s[k] ? k * 2 : k * 2 + 1;//关闭时小球落在左节点，反之落在右节点；
				if (k > n) break;//出界；
			}
		}
		printf("%d\n", k / 2);//出界之前的叶子编号；
	}
	return 0;
}