给定一个N叉树,找到其最大深度。
最大深度是指从根节点到最远叶子节点的最长路径上的节点总数。
例如,给定一个 3叉树 :
我们应返回其最大深度,3。
说明:
树的深度不会超过 1000。
树的节点总不会超过 5000。
题解:
#include <iostream>
#include <vector>
using std::cout;
using std::endl;
using std::vector;
using std::system;
struct Node {
Node(int value) : value_(value) {}
Node(int value, vector<Node*>& children) : value_(value), children_(children) {}
int value_;
vector<Node*> children_;
};
size_t MaxDepth(const Node* root);
int main(int argc, const char* argv[]) {
Node* root = new Node(1);
root->children_.push_back(new Node(3));
root->children_.push_back(new Node(2));
root->children_.push_back(new Node(4));
root->children_[1]->children_.push_back(new Node(5));
root->children_[1]->children_.push_back(new Node(6));
cout << MaxDepth(root) << endl;
system("pause");
return 0;
}
size_t MaxDepth(const Node* root) {
if (root == nullptr)
return 0;
// 不撞南墙不回头
size_t max_depth = 0;
for (decltype(root->children_.size()) i = 0; i != root->children_.size(); ++i) {
size_t depth = MaxDepth(root->children_[i]);
if (depth > max_depth)
max_depth = depth;
}
return max_depth + 1;
}