树的遍历算法(递归,迭代)(C++,java)
最近刷力扣,记录一下树的各种遍历算法,分别用C++和java实现,基于递归和迭代。
基于递归的二叉树遍历
所有树的遍历用递归来解决思路都是是十分清晰的,前序中序后序差距都差不多,就是改变一下位置。下面给出前序的代码。
- 前序 (当前节点+左子树+右子树)
//递归(java)
public List<Integer> preorderTraversal(TreeNode root) {
List<Integer> list = new ArrayList<>();
if (root!=null){
list.add(root.val);
list.addAll(preorderTraversal(root.left));
list.addAll(preorderTraversal(root.right));
}
return list;
}
//递归(C++)
public:
vector<int> result;
vector<int> preorderTraversal(TreeNode* root) {
if(root){
this.result.push_back(root->val);
preorderTraversal(root->left);
preorderTraversal(root->right);
}
return result;
}
-
中序 (左子树+当前节点+右子树)
-
后序 (左子树+右子树+当前节点)
基于迭代的二叉树遍历
基于迭代的二叉树遍历实现起来要稍微复杂一点,需要利用栈来实现,为了防止指针丢失,将迭代到的每一个节点都存入栈中。可以用以下模板。
while(){
while()
}
- 前序
//java
public List<Integer> preorderTraversal(TreeNode root) {
List<Integer> list = new ArrayList<>();
Stack<TreeNode> stack = new Stack<>();
while(root!=null||!stack.isEmpty()){
while(root!=null){
list.add(root.val);
stack.push(root);
root = root.left;
}
root = stack.pop()