輸入二叉樹數組生成對應的二叉樹資料結構（JavaScript） 需求

需求

• 輸入一個二叉樹的數組，函數能夠生成對應的二叉樹的結構

代碼

const deserialize = (data) => {

    let res = ''
    for (let v of data) {
        res = res + v + ',';
    }
    for (let i = 0; i <= data.length; i++) {
        res = res + 'null' + ','
    }
    data = res;

    function TreeNode(val) {
        this.val = val;
        this.left = null;
        this.right = null;
    }
    if (data.length === 0) return null;
    const list = data.split(',');   // split成數組
    list.splice(list.length - 1);
    let list_Pointer = 1;
    let queue_pointer = 0;
    const root = new TreeNode(list[0])
    let queue = [root];

    while (list_Pointer !== list.length) {
        if (queue[0] === null) {
            queue.shift();
            queue_pointer++;
            continue;
        }
        if (queue_pointer === list_Pointer) {
            list_Pointer = list_Pointer + 2;
        }
        if (list[list_Pointer] === 'null') {
            queue[0].left = null;
        } else {
            queue[0].left = new TreeNode(list[list_Pointer]);
        }
        if (list[list_Pointer + 1] === 'null') {
            queue[0].right = null;
        } else {
            queue[0].right = new TreeNode(list[list_Pointer + 1]);
        }
        queue.push(queue[0].left);
        queue.push(queue[0].right);
        queue.shift();
        queue_pointer++
        list_Pointer = list_Pointer + 2;
    }
    return root
};

const test2 = deserialize([1,2,3,4]);
      

注意事項

這裡的二叉樹的每一個節點的val值不是Number類型的，而是字元串類型的