有一棵二叉树,最大深度为D,且所有叶子的深度都相同。所有结点从上到下从左到右编号为1,2,3,...,2^D-1。在结点1处放一个小球,它会往下落。每个内结点上都有一个开关,初始全部关闭,当每次有小球落到一个开关上时,它的状态都会改变。当小球到达一个内结点时,如果该结点上的开关关闭,则往左走,否者往右走,直到走到叶子结点。
一些小球从结点1处依次开始下落,最后一个小球将会落到哪里呢?输入叶子深度D和小球个数I,输出第I个小球最后所在的叶子编号。假设I不超过整棵树的叶子个数(即2^(D-1))。D<=20。输入最多包含1000组数据。
样例输入: 样例输出:
4 2 12
3 4 7
10 1 512
2 2 3
8 128 255
16 12345 36358
对于任意一个父节点来看,左子树的节点号为2*k,右子树节点号为2*k+1。初始状态所有节点关闭,第一个球则默认从左下落。
//代码参照白书
#include <stdio.h>
#include <string.h>
#define max 20
int s[1<<max];//最大节点个数为2^max-1,s[0]不算入节点
int main()
{
int d,i,j;
while(~scanf("%d%d",&d,&i))
{
memset(s,0,sizeof(s));//开关
int k,n=(1<<d)-1;//n是最大节点编号
for(j=0;j<i;j++)//连续让i个球下落
{
k=1;
while(k<=n) //k>n已经落“出界”了
{
s[k]=!s[k];
k=s[k]?k*2:k*2+1; //根据开关状态选择下落方向
}
}
printf("%d\n",k/2); //出界之前的叶子编号
}
return 0;
}
以上常规解法数据量巨大,测试数据下落层数最大可以达到2^19*19,找规律可以发现,小球最终落入叉树,必然一个进入左子树一个进入右子树,进入情况与编号奇偶性有关。I为奇数时,当前小球即为第(I+1)/2个小球,I为偶数即为右走第I/2个小球.
//代码参照白书的改进代码
#include <stdio.h>
int main()
{
int d,i,j,k;
while(~scanf("%d%d",&d,&i))
{
k=1;
for(j=0;j<d-1;j--)
{
if(i%2)
{
k*=2;
i=(i+1)/2;
}
else
{
k=k*2+1;
i/=2;
}
}
printf("%d\n",k);
}
return 0;
}