二叉树相关遍历

// 二叉树前序遍历
function loop(root) {
    if(!root) return;
    console.log(root.value);
    loop(root.left);
    loop(root.right);
}

// 二叉树广度遍历(自上而下)
function levelLoop(root) {
    if(!root) return [];
    const queue = [root];
    const results = [];
    while(queue.length) {
        let len = queue.length;
        const currValueList = []
        while(len--) {
            const cur = queue.shift();
            currValueList.push(cur.value);
            cur.left && queue.push(cur.left);
            cur.right && queue.push(cur.right)
        }
        results.push(currValueList)
    }
    return results;
    }

 // 二叉树广度遍历(自下而上)
function levelLoop(root) {
    if(!root) return [];
    const queue = [root];
    const results = [];
    while(queue.length) {
        let len = queue.length;
        const currValueList = []
        while(len--) {
            const cur = queue.shift();
            currValueList.push(cur.value);
            cur.left && queue.push(cur.left);
            cur.right && queue.push(cur.right)
        }
        results.unshift(currValueList)  // 只修改这一条即可
    }
    return results;
}

// 之字形遍历二叉树
function levelLoop(root) {
    if(!root) return [];
    const res = []
    const queue = [root];
    let level = 1; // 用level控制从哪个方向push
    while(queue.length) {
        let len = queue.length;
        const curValueList = [];
        while(len--) {
            let cur = queue.shift();
            level % 2 === 0 ? curValueList.unshift(cur.value): curValueList.push(cur.value);
            cur.left && queue.push(cur.left);
            cur.right && queue.push(cur.right);
        }
        level++;
        res.push(curValueList);
    }
    return res;
}


// 遍历N叉树
function levelLoop(root) {
    if(!root) return [];
    const res = []
    const queue = [root];
    while(queue.length) {
        let len = queue.length;
        const curValueList = [];
        while(len--) {
            let cur = queue.shift();
            curValueList.push(cur.value);
            queue.push(...cur.children); // 将所有的孩子push进去
        }
        res.push(curValueList);
    }
    return res;
}

// 二叉树右视图(写法一)
function levelLoop(root) {
    if(!root) return [];
    const res = []
    const queue = [root];
    while(queue.length) {
        let len = queue.length;
        const curValueList = [];
        while(len--) {
            let cur = queue.shift();
            curValueList.push(cur.value);
            cur.left && queue.push(cur.left); 
            cur.left && queue.push(cur.left);
        }
        res.push(curValueList.pop()); // 只收集每一层的最后一个
    }
    return res;
}

// 二叉树右视图(写法二)
function levelLoop(root) {
    if(!root) return [];
    const res = []
    const queue = [root];
    while(queue.length) {
        let len = queue.length;
        let curValue;
        while(len--) {
            let cur = queue.shift();
            curValue = cur.value;
            cur.left && queue.push(cur.left); 
            cur.left && queue.push(cur.left);
        }
        res.push(curValue); // 只收集每一层的最后一个
    }
    return res;
}



评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值