小猴子下落
时间限制: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 ", 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 }