nyoj 63 小猴子下落

小猴子下落

时间限制：3000 ms  |  内存限制：65535 KB

难度：3

描述

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

一些小猴子从结点1处开始往下跑，最后一个小猴儿会跑到哪里呢？

输入

输入二叉树叶子的深度D,和小猴子数目I，假设I不超过整棵树的叶子个数，D<=20.最终以 0 0 结尾

输出

输出第I个小猴子所在的叶子编号。

样例输入

4 2
3 4
0 0

样例输出

12
7

分析：网上有优化解法，我还是用的dfs：

 1 #include <iostream>
 2 #include <cstdio>
 3 #include <cstring>
 4 using namespace std;
 5
 6 bool a[1000010];
 7 int D, I;
 8
 9 void dfs(int depth, int id, int num){
10     if (depth == D) {
11         if(num == I)
12             printf("%d\n", id);
13         return;
14     }
15     if (a[id] == false) {
16         a[id] = true;
17         dfs(depth + 1, id << 1, num);
18     } else if (a[id] == true) {
19         a[id] = false;
20         dfs(depth + 1, (id << 1) + 1, num);
21     }
22 }
23
24 int main(){
25     while(scanf("%d%d", &D, &I)){
26         if(D == 0 && I == 0)
27             break;
28         memset(a, false, sizeof(a));
29         for(int i = 1; i <= I; i++){
30             dfs(1, 1, i);
31         }
32     }
33     return 0;
34 }