Leetcode 515. Find Largest Value in Each Tree Row

https://leetcode.com/problems/find-largest-value-in-each-tree-row/description/

You need to find the largest value in each row of a binary tree.

Example:

Input: 

          1
         / \
        3   2
       / \   \  
      5   3   9 

Output: [1, 3, 9]

按深度遍历树,最直观的就是bfs,利用队列来一层一层的进行遍历了。

var largestValues = function(root) {
    const rowArr = [];
    if (!root) {
        return rowArr;
    }
    const queue = [root];
    while (queue.length > 0) {
        const len = queue.length;
        let max = queue[0].val; // queue is not empty then there must be at least one element here
        for (let i = 0; i < len; ++i) {
            const current = queue.shift();
            max = current.val > max ? current.val : max;
            if (current.left) {
                queue.push(current.left);
            }
            if (current.right) {
                queue.push(current.right);
            }
        }
        rowArr.push(max);
    }
    return rowArr;
};

除了bfs,我们也可以利用递归dfs来遍历,只要传入一个当前层数作为标记,并传入一个每层的结果数组。将每层的结果保存在rowArr中,最后在对每层结果进行处理,返回每层的最大值。

var largestValues = function(root) {
    const rowArr = [];
    getRows(root, 0, rowArr);
    return rowArr.map((arr)=>{
        return arr.reduce((pre, next) => {
            return pre > next ? pre : next;
        }, arr[0]);
    });
};

const getRows = (root, rowNum, rowArr) => {
    if (root) {
        if (rowNum >= rowArr.length) {
            rowArr.push([]);
        }
        rowArr[rowNum].push(root.val);
        getRows(root.left, rowNum+1, rowArr);
        getRows(root.right, rowNum+1, rowArr);
    }
};

这样就beat 70%了,要是在递归中直接把最大值处理返回了会怎样呢?

var largestValues = function(root) {
    const rowArr = [];
    getRows(root, 0, rowArr);
    // return rowArr.map((arr)=>{
    //     return arr.reduce((pre, next) => {
    //         return pre > next ? pre : next;
    //     }, arr[0]);
    // });
    return rowArr;
};

const getRows = (root, rowNum, rowArr) => {
    if (root) {
        if (rowNum >= rowArr.length) {
            rowArr.push(root.val);
        } else {
            rowArr[rowNum] = rowArr[rowNum] > root.val ? rowArr[rowNum] : root.val;
        }
        getRows(root.left, rowNum+1, rowArr);
        getRows(root.right, rowNum+1, rowArr);
    }
};

结果是beat 40%,估计是由于javascript数组的查询修改比直接后面push一个要更加耗时吧。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
经导师精心指导并认可、获 98 分的毕业设计项目!【项目资源】:微信小程序。【项目说明】:聚焦计算机相关专业毕设及实战操练,可作课程设计与期末大作业,含全部源码,能直用于毕设,经严格调试,运行有保障!【项目服务】:有任何使用上的问题,欢迎随时与博主沟通,博主会及时解答。 经导师精心指导并认可、获 98 分的毕业设计项目!【项目资源】:微信小程序。【项目说明】:聚焦计算机相关专业毕设及实战操练,可作课程设计与期末大作业,含全部源码,能直用于毕设,经严格调试,运行有保障!【项目服务】:有任何使用上的问题,欢迎随时与博主沟通,博主会及时解答。 经导师精心指导并认可、获 98 分的毕业设计项目!【项目资源】:微信小程序。【项目说明】:聚焦计算机相关专业毕设及实战操练,可作课程设计与期末大作业,含全部源码,能直用于毕设,经严格调试,运行有保障!【项目服务】:有任何使用上的问题,欢迎随时与博主沟通,博主会及时解答。 经导师精心指导并认可、获 98 分的毕业设计项目!【项目资源】:微信小程序。【项目说明】:聚焦计算机相关专业毕设及实战操练,可作课程设计与期末大作业,含全部源码,能直用于毕设,经严格调试,运行有保障!【项目服务】:有任何使用上的问题,欢迎随时与博主沟通,博主会及时解答。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值