LeetCode算法问题17 —— Find Largest Value in Each Tree Row

首先看一下问题描述



要求我们找到一个二叉树每一层的最大值

这个问题用BFS、DFS都可以解决，且不得不遍历整棵树。在遍历过程中，需要时刻记录当前节点的深度，以便比较同深度下的节点值以获取每一层的最大值

本题我采用了DFS算法，对于主干函数largestValues，通过操作：



通过explore函数对容器answer进行填满。对于容器answer，answer[i]表示深度为i的一层(起始深度为0)的最大值。

对于explore函数

void explore(TreeNode* node, int depth, vector<int>& valueOfEachRow){}

参数有各自的定义：

node：当前遍历到的节点

depth：该节点在树中的深度

valueOfEachRow：用于记录每层最大值的容器

以此可通过DFS完成对每层最大值的提取。

具体代码如下：

void explore(TreeNode* node, int depth, vector<int>& valueOfEachRow) {
if (node == NULL)
return;
if (valueOfEachRow.size() <= depth) {
valueOfEachRow.push_back(node->val);
} else {
if (valueOfEachRow[depth] < node->val)
valueOfEachRow[depth] = node->val;
}
explore(node->left, depth + 1, valueOfEachRow);
explore(node->right, depth + 1, valueOfEachRow);
}

vector<int> largestValues(TreeNode* root) {
vector<int> answer;
explore(root, 0, answer);
return answer;
}