Binary Tree Inorder Traversal

题目描述:

Given a binary tree, return the inorder traversal of its nodes' values.

For example:
Given binary tree {1,#,2,3},

   1
    \
     2
    /
   3

return [1,3,2].

Note: Recursive solution is trivial, could you do it iteratively?

这个递归方法比较简单,代码如下:

public List<Integer> inorderTraversal(TreeNode root) {
    List<Integer> list = new ArrayList<Integer>();
    getInorderTraversal(root, list);
    return list;
}

public void getInorderTraversal(TreeNode root,List<Integer> list){
	if(root==null)
		return;
	getInorderTraversal(root.left, list);
	list.add(root.val);
	getInorderTraversal(root.right, list);
}

但是这个题目要求用迭代,用一个set记录已经被访问过的左孩子节点就可以了。代码如下:

public List<Integer> inorderTraversal(TreeNode root) {
    Set<TreeNode> visited=new HashSet<TreeNode>();
    List<Integer> result=new ArrayList<Integer>();
    Stack<TreeNode> stack=new Stack<TreeNode>();
    stack.push(root);
    while(!stack.isEmpty()){
    	TreeNode p=stack.peek();
    	if(p.left!=null&&!visited.contains(p.left)){
    		stack.push(p.left);
    	}else{
    		p=stack.pop();
    		result.add(p.val);
    		visited.add(p);
    		if(p.right!=null){
    			stack.add(p.right);
    		}
    	}
    }
    return result;
}


阅读更多
版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/yeshiwu/article/details/51555052
个人分类: leetcode
上一篇Palindrome Pairs
下一篇Unique Binary Search Trees II
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

关闭
关闭