LeetCode刷題助手：根據數組生成二叉樹

大家在刷LeetCode的二叉樹相關的題目的時候，遇到問題需要在本地IDE裏調試的時候，題目裏給的二叉樹的形式是一個數組，類似

Integer[] array = new Integer[]{1, null, 2, null, null, null, 3};

這種形式，它是以null來做填充，構成一個完全二叉樹，具體的圖我就不畫了，大家自己理解一下。而我們需要自己手動把它轉換成二叉樹的形式，再把這個二叉樹作爲參數來進行解題。

因爲感覺這樣做十分的麻煩，所以我寫了一個轉換方法，可以把上面這種類型的數組自動轉化爲一個二叉樹。

首先定義二叉樹節點：

class TreeNode {
    int val;
    TreeNode left;
    TreeNode right;
    TreeNode(int x) { val = x; }
}

轉換方法如下：

    public static TreeNode convert(Integer[] array) {
        int floor = 0, count = 0;
        TreeNode[] treeNodes = new TreeNode[array.length];
        while (array != null && count < array.length) {
            int start = (int) Math.pow(2, floor) - 1;
            int end = (int) Math.pow(2, floor + 1) - 1;
            if (end > array.length) {
                end = array.length;
            }
            for (int i = start; i < end; i++) {
                if (array[i] != null) {
                    treeNodes[i] = new TreeNode(array[i]);
                } else {
                    treeNodes[i] = null;
                }
                if (i > 0) {
                    int parent = (i - 1) / 2;
                    if (parent >= 0) {
                        TreeNode pNode = treeNodes[parent];
                        if (pNode != null) {
                            if (i % 2 == 1) {
                                pNode.left = treeNodes[i];
                            } else {
                                pNode.right = treeNodes[i];
                            }
                        }
                    }
                }
                count++;
            }
            floor++;
        }
        return treeNodes[0];
    }

測試代碼：

    public static void main(String[] args) {
        Integer[] array = new Integer[]{1, null, 2, null, null, null, 3};
        TreeNode root = convert(array);

    }

用debug查看結果：

我沒有寫特別多的測試用例，所以用的時候如果有問題，麻煩自己調試修改一下。