首页 > 代码库 > nyoj 63 小猴子下落

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 23 40 0
样例输出
127

分析:网上有优化解法,我还是用的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 }

 

 

nyoj 63 小猴子下落