一、题意
Given a binary tree, return the inorder traversal of its nodes’ values.
For example:
Given binary tree [1,null,2,3],
1
\
2
/
3
return [1,3,2].
二、分析和解答
其实就是中序遍历,无非是把visit()具体化成了list数组。
1、递归
void inorder(TreeNode node,List list){
if(node != null){
inorder(node.left,list);
list.add(node.val);
inorder(node.right,list);
}
}
public List<Integer> inorderTraversal(TreeNode root) {
List<Integer> list = new ArrayList();
if(root == null)
return list;
inorder(root,list);
return list;
}
Note:我在一个地方卡了很长时间,逻辑上都没有错误:
(1)list.add(node.val);
一定注意add的是node.val值,而不是node,之前我已经错过一次了,如果是是node的话,那将会是一个链表的链表。
(2)当root为NULL的时候,返回list,而不能返回NULL;
(3)递归的时候,一般不返回值,当然非要返回也行,课本上没写。
2、非递归
非递归的中序遍历背了多少遍了,还是记不住,看来还是没有真正的掌握啊!后来想了一会,又想起来了!试了一下,果然对,外加一些小心得!
public List<Integer> inorderTraversal(TreeNode root) {
List<Integer> list = new ArrayList();
Stack<TreeNode> stack = new Stack();
if(root == null)
return list;
TreeNode cur = root;
while(cur!=null || !stack.empty()){
if(cur != null){
stack.push(cur);
cur = cur.left;
}else if(!stack.empty()){
cur = stack.pop();
list.add(cur.val);
cur = cur.right;
}
}
return list;
}
Note:
(1)首先,申请一个指针指向根节点!循环条件就是p不为空或者栈不为NULL。
(2)while循环中的两个判断的关系是if 和 else if 的关系,不能写错。或者还有另外一种写法也对:
public List<Integer> inorderTraversal(TreeNode root) {
List<Integer> list = new ArrayList();
Stack<TreeNode> stack = new Stack();
if(root == null)
return list;
TreeNode cur = root;
while(cur!=null || !stack.empty()){
while(cur != null){
stack.push(cur);
cur = cur.left;
}
if(!stack.empty()){
cur = stack.pop();
list.add(cur.val);
cur = cur.right;
}
}
return list;
}
(3)什么时候栈为空呢?当根节点从栈中弹出来的时候,栈变为null,cur又指向根节点的右孩子;由于当前cur不为NULL,所以又一次进入了循环。