//在二叉树中寻找值最大的节点并返回。
输入:
{1,-5,3,1,2,-4,-5}
输出: 3
说明:
这棵树如下所示:
1
/ \
-5 3
/ \ / \
1 2 -4 -5
1.通过递归遍历整个树结构
获取根节点,递归比较左右节点的大小。
2.通过将树压入栈,弹栈获取树种最大元素
public static TreeNode maxNode1(TreeNode root) {
if(root==null)
return null;
if(root.left == null && root.right == null)
return root;
TreeNode max = root;
Stack<TreeNode> s = new Stack<>();
System.out.println(root);
s.push(root);
while(s.size()!=0){
System.out.println(s.size());
TreeNode temp = s.pop();
System.out.println("栈顶元素 " + temp);
if(temp.left!=null) {
System.out.println("左节点 " +temp.left);
s.push(temp.left);
}
if(temp.right!=null) {
System.out.println("右节点 " +temp.right);
s.push(temp.right);
}
if(temp.val>max.val) {
System.out.println(" 栈顶弹出元素 "+temp + " 当前最大元素 "+ max);
max = temp;
}
}
return max;
}
public static void main(String[] args) {
TreeNode treeNode = new TreeNode(1);
TreeNode treeNode1 = new TreeNode(-5);
treeNode.left = treeNode1;
TreeNode treeNode2 = new TreeNode(3);
treeNode1.left = treeNode2;
TreeNode treeNode3 = new TreeNode(1);
treeNode1.right = treeNode3;
TreeNode treeNode4 = new TreeNode(2);
treeNode.right = treeNode4;
TreeNode treeNode5 = new TreeNode(-4);
treeNode4.left = treeNode5;
TreeNode treeNode6 = new TreeNode(-5);
treeNode4.right = treeNode6;
TreeNode treeNode7 = maxNode1(treeNode);
System.out.println(treeNode7);
}
public class TreeNode {
public int val;
public TreeNode left, right;
public TreeNode(int val) {
this.val = val;
this.left = this.right = null;
}
}
- 将根节点压入栈。
- 赋值给最大节点变量。
- 判断栈是否为空。
- 不为空将该元素弹栈,赋值给临时变量。
- 判断该元素的左右子树是否为空。
- 不为空将左右子树压入栈。
- 为空无操作
- 判断临时节点与最值节点大小关系,大于最大值节点更新最大值节点。
- while条件是否为true?为true继续遍历。不为true返回最大节点。
上图是栈内数据变化过程。