许多的小球一个一个的从一棵满二叉树上掉下来组成FBT(Full Binary Tree,满二叉树),每一时间,一个正在下降的球第一个访问的是非叶子节点。然后继续下降时,或者走右子树,或者走左子树,直到访问到叶子节点。决定球运动方向的是每个节点的布尔值。最初,所有的节点都是FALSE,当访问到一个节点时,如果这个节点是FALSE,则这个球把它变成TRUE,然后从左子树走,继续它的旅程。如果节点是TRUE,则球也会改变它为FALSE,而接下来从右子树走。满二叉树的标记方法如下图。
因为所有的节点最初为FALSE,所以第一个球将会访问节点1,节点2和节点4,转变节点的布尔值后在在节点8停止。第二个球将会访问节点1、3、6,在节点12停止。明显地,第三个球在它停止之前,会访问节点1、2、5,在节点10停止。 现在你的任务是,给定FBT的深度D,和I,表示第I个小球下落,你可以假定I不超过给定的FBT的叶子数,写一个程序求小球停止时的叶子序号。
输入文件仅一行包含两个用空格隔开的整数D和I。其中2<=D<=20,1<=I<=524288。
对应输出第I个小球下落停止时的叶子序号。
4 2
12
#include "bits/stdc++.h"
#define lc number << 1
#define rc number << 1 | 1
using namespace std;
struct node{
int num, l, r;
bool direct = 0;
} tree[1050000];
int D, I, ans;
void build(int n){
for (int i = 1; i <= (1 << n) - 1; i++){
int number = tree[i].num;
tree[i].num = i;
if (i*2+1<=(1<<n)-1){
tree[i].l = lc;
tree[i].r = rc;
}
else{
tree[i].l = 0;
tree[i].r = 0;
}
}
}
void find1(int x){
if(tree[x].direct == 1){
tree[x].direct = 0;
find1(tree[x].l);
}
else{
tree[x].direct = 1;
find1(tree[x].r);
}
return ;
}
int query(int tot){
for (int i = 1; i <= tot; i++){
if (tree[i].l == 0 && tree[i].r == 0)
ans = tree[i].num;
else{
find1(1);
}
}
return ans;
}
int main(){
ios::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
cin >> D >> I;
build(D);
cout << query(I);
return 0;
}
但是我的代码答案有问题,求大佬修改