Given a binary tree, find its maximum depth.
The maximum depth is the number of nodes along the longest path from the root node down to the farthest leaf node.
Solution:
Recursive:
/**
* Definition for binary tree
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
public class Solution {
public int maxDepth(TreeNode root) {
if (root == null)
return 0;
return Math.max(maxDepth(root.left), maxDepth(root.right))+1;
}
}
Iterative:
/**
* Definition for binary tree
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
public class Solution {
public int maxDepth(TreeNode root) {
if(root == null)
return 0;
int max=0;
Deque<TreeNode> nodeStack = new ArrayDeque<TreeNode>();
Deque<Integer> depthStack = new ArrayDeque<Integer>();
nodeStack.addFirst(root);
depthStack.addFirst(1);
while(!nodeStack.isEmpty())
{
TreeNode node = nodeStack.removeFirst();
int depth = depthStack.removeFirst();
max = depth>max ? depth : max;
if (node.left != null)
{
nodeStack.addFirst(node.left);
depthStack.addFirst(depth+1);
}
if(node.right != null)
{
nodeStack.addFirst(node.right);
depthStack.addFirst(depth+1);
}
}
return max;
}
}
No comments :
Post a Comment