【LeetCode】144. Binary Tree Preorder Traversal

最新推荐文章于 2024-05-30 07:40:02 发布

Juxin_Lin

最新推荐文章于 2024-05-30 07:40:02 发布

阅读量376

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode 文章标签： leetcode

本文链接：https://blog.youkuaiyun.com/Juxin_Lin/article/details/52453842

leetcode 专栏收录该内容

39 篇文章

订阅专栏

本文介绍了一种解决二叉树前序遍历的方法，包括递归和非递归两种实现方式。递归方法简单直观，而非递归方法则通过栈来模拟访问过程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Difficulty: Medium

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

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

return [1,2,3].

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

解法一：使用递归求解，判断当前根节点是否为空，先遍历当前子树的根节点，然后递归遍历当前节点的左子树，然后递归遍历当前节点的右子树。

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    vector<int> preorderTraversal(TreeNode* root) {
        if(root)
        {
        	v.push_back(root->val);
        	preorderTraversal(root->left);
        	preorderTraversal(root->right);
        }
        
        return v;
    }
    
private:
    vector<int> v;
};

一开始，我没有多想，一下子就把if那个条件判断写成了while，然后就TLE了，所以做题还是要想清楚每一步是怎么样的，不要想当然。

解法二：使用非递归的做法，先判断根节点是否为空，然后进行模拟先序访问，如果左子树非空，并且之前没有访问过，就访问左子树，然后访问右子树，也同样是要判断之前有没有访问过，每次访问将节点入栈，如果左子树和右子树都空，或者都访问过了，那么出栈。栈为空时访问结束。

class Solution {
public:
    vector<int> preorderTraversal(TreeNode* root) {
    	vector<int> v;
    	if(root==NULL)
    	return v;
    	else{
    		stack<TreeNode*> proc;
    	proc.push(root);
    	v.push_back(root->val);
    	map<TreeNode*,int> visit;
    	visit[root]=1;
    	while(!proc.empty())
    	{
    		TreeNode* node = proc.top();
    	
    		if(node->left!=NULL && visit[node->left]!=1)
    		{
    			node=node->left;
    			visit[node]=1;
    			v.push_back(node->val);
    			proc.push(node);
    		}
    		else if(node->right!=NULL && visit[node->right]!=1)
    		{
    			node=node->right;
    			visit[node]=1;
    			v.push_back(node->val);
    			proc.push(node);
    		}
    		else{
    			proc.pop();
    		}
    	}
    	}
    	
        return v;
    }
    
    
};