小球下落 二叉树

小球下落 （二叉树的应用）

有一棵二叉树，最大深度为D，且所有叶子的深度都相同。所有结点从上到下从左到右编号为1，2，3，...，2^D-1。在结点1处放一个小球，它会往下落。每个内结点上都有一个开关，初始全部关闭，当每次有小球落到一个开关上时，它的状态都会改变。当小球到达一个内结点时，如果该结点上的开关关闭，则往左走，否者往右走，直到走到叶子结点。
一些小球从结点1处依次开始下落，最后一个小球将会落到哪里呢？输入叶子深度D和小球个数I，输出第I个小球最后所在的叶子编号。假设I不超过整棵树的叶子个数（即2^(D-1)）。D<=20。输入最多包含1000组数据。
样例输入：
4 2
3 4
10 1
2 2
8 128
16 12345
样例输出：
12
7
512
3
255
36358

[分析]
不难发现，对于一个结点k，它的左儿子、右儿子的编号分别是2k和2k+1 。所以方法[1] 可以模拟程序输出结果。但是这种方法的缺点是：运算量太大 ，由于小球个数I可以高达2^(D-1)，每个测试数据下落总层数可能会高达2^19*(20-1)=9961472。所以效率比较低。

考虑方法[2] ，每个小球都会落在根结点上，因此前两个小球必然是一个在左子树，一个在右子树。一般地，只需看小球的编号的奇偶性，就能知道它是最终在哪棵子树中 。对于那些落入根结点左子树的小球来说，只需知道该小球是第几个落在根的左子树里的，就可以知道它下一步往左还是往右了。以此类推，直到小球落到叶子上。当e是奇数时，它是往左走的第(e+1)/2个小球；当I是偶数时，它是往右走的第e/2个小球。这样，可以直接模拟最后一个小球的路线 。这样，程序的运算量就与小球编号无关 了，而且节省了一个巨大的数组 tree，程序的效率很高。

方法一代码：

#include <iostream>
#include<string.h>
using namespace std;
const int MAX=20;
int a[1<<MAX];  //也就是a[2的20次方]

int main()
{
    int d,e,k;

    while(cin>>d>>e)
    {

      memset(a,0,sizeof(a));

       int n=(1<<d)-1;
        for(int i=1; i<=e; i++)
        {
            k=1;
            while(k<=n)
            {
                if(a[k]==0)k=k*2;
                else  k=k*2+1;
                a[k/2]=!a[k/2];  //注意k的值已经变了，让a[k]转换的时候别弄错了
              }
         }
            cout<<k/2<<endl;
      }
        return 0;
    }

方法二代码：

#include <iostream>
#include<string.h>
using namespace std;

int main()
{
    int d,e,k;
    while(cin>>d>>e)  //d用来记录二叉树的深度，e来表示第e个小球同时也表示第e次在根节点个节点
    {
        k=1;
        for(int i=1;i<=d-1;i++)
        {
            if(e%2==0)
            {
                k=k*2+1;
                e/=2;  //当k=上面的那个数时即（k*2+1），
            }
            else
            {
                k=k*2;
                e=(e+1)/2;
            }
         }
        cout<<k<<endl;
    }
        return 0;
    }