小球下落 Dropping Balls, UVa 679

题目描述

有一棵二叉树，最大深度为D，且所有的叶子深度都相同。所有结点从上到下从左到右编号为1，2，3，…，2eD-1。在结点1处放一个小球，它会往下落。每个结点上都有一个开关，初始全部关闭，当每次有小球落到一个开关上时，它的状态都会改变。当小球到达一个内结点时，如果该结点的开关关闭，则往上走，否则往下走，直到走到叶子结点，如下图所示。

一些小球从结点1处依次开始下落，最后一个小球将会落到哪里呢？输入叶子深度D和小球个数I，输出第I个小球最后所在的叶子编号。假设I不超过整棵树的叶子数；D<=20。输出最多包含1000组数据。

样例输入

4 2
3 4
10 1
2 2
8 128
16 12345

样例输出

12
7
512
3
255
36358

分析
给定一棵包含2的d次方个结点（其中d为树的高度）的完全二叉树，如果把结点从上到下从左到右编号为1，2，3…，则对于一个结点k，其左子节点、右子节点的编号分别是2k和2k+1

模拟实现

import java.util.Scanner;

public class DroppingBalls {

    public static void main(String[] args) {
        int D;// 叶子深度
        int I;// 小球个数
        boolean[] s;// 结点开关状态
        Scanner sc = new Scanner(System.in);
        while (sc.hasNext()) {
            D = sc.nextInt();
            I = sc.nextInt();
            s = new boolean[1 << D];
            for (int i = 0; i < s.length; i++) {
                s[i] = false;
            }
            int k = 1;
            int n = (1 << D) - 1;// 最大结点编号
            for (int i = 0; i < I; i++) {// 连续让I个小球下落
                k = 1;
                for (;;) {
                    s[k] = !s[k];
                    k = s[k] ? k * 2 : k * 2 + 1;// 根据开关状态选择下落方向
                    if (k > n)// 已经落出界了
                        break;
                }
            }
            System.out.println(k / 2);// 出界之前的叶子编号
        }
        sc.close();
    }

}

在本题中，每个小球都是严格下落D-1层，采用上述模拟的方法，结点数是2的D次方-1，每组测试数据下落总层数可能会过大，运算量太大，所以要采取优化。
每个小球都会落在根结点上，因此前两个小球必然是一个在左子树，一个在右子树。一般地，只需看小球编号的奇偶性，就能知道它最终在哪棵子树中。对于那些落入根结点左子树的小球来说，只需知道该小球是第几个落在根的左子树里的，就可以知道它下一步往左还是往右了。以此类推，直到小球落到叶子上。
当I是奇数时，它是往左走的第（I+1）/2个小球；当I是偶数时，它是往右走的第I/2个小球。这样，可以直接模拟最后一个小球的路线。这样，程序的运算量就与小球编号无关了，而且节省一个巨大的s数组。

优化版本

public class DroppingBalls2 {
    public static void main(String[] args) {
        int D;// 叶子深度
        int I;// 小球个数
        Scanner sc = new Scanner(System.in);
        while (sc.hasNext()) {
            D = sc.nextInt();
            I = sc.nextInt();
            int k = 1;
            for (int i = 0; i < D - 1; i++) {
                if (I % 2 == 1) {
                    k = k * 2;
                    I = (I + 1) / 2;
                } else {
                    k = k * 2 + 1;
                    I /= 2;
                }
            }
            System.out.println(k);
        }
        sc.close();
    }
}