LeetCode 94. Binary Tree Inorder Traversal

博客围绕二叉树中序遍历展开,提到递归解法易想但速度慢,还介绍了迭代解法,思路是先找最左侧结点并将路径点入栈,再删去指向左子点的指针。最后还建议做完本题后尝试144的前序遍历。

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

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

Example:

Input: [1,null,2,3]
   1
    \
     2
    /
   3

Output: [1,3,2]

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

题目给出了定义好的树结构:

struct TreeNode {
	int val;
	TreeNode *left;
	TreeNode *right;
	TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};

这道题的递归解法比较用以想到,但解题会比较慢:

递归实现的代码:

class Solution {		//递归方法
public:
	vector <int>vec;
	void find(TreeNode *root) {
		if(root != NULL) {
			find(root->left);
			vec.push_back(root->val);
			find(root->right);
		}
	}
	vector<int> inorderTraversal(TreeNode* root) {
		find(root);
		return vec;
	}
};

看了大佬的栈实现,受益匪浅,试了一下,果然厉害!题目也说了最好用迭代试一下。迭代的思路,是先找到整棵树最左侧的结点,将一路找到的点都放栈中,然后删掉该点指向左子点的指针(思考一下应该会明白其中的道理)。

class Solution1		//迭代方法
{
public:
	Solution1();
	~Solution1();
	stack<TreeNode*> st;
	vector<int> inorderTraversal(TreeNode* root) {
		vector<int> ans;
		if (root != NULL) {
			st.push(root);
			TreeNode *troot;
			while (!st.empty()) {        //当栈里还有元素,继续循环
				root = st.top();
				st.pop();
				while (root != NULL) {
					troot = root;
					root = root->left;  //记得要把该点指向左子点的指针置空
					troot->left = NULL;
					st.push(troot);
				}
				root = st.top();
				st.pop();
				ans.push_back(root->val);
				//cout << root->val << endl;
				root = root->right;
				if (root != NULL) {
					st.push(root);
				}

			}
		}
		
		return ans;
	}


private:

};

结果也很感人!

做完这道题,可以立刻写一下144的前序遍历

class Solution2 {
public:
	stack<TreeNode*> st;
	map<TreeNode*,int> hash;
	vector<int> preorderTraversal(TreeNode* root) {
		vector<int> ans;
		TreeNode *lr;
		if (root) {
			st.push(root);
			while (!st.empty()) {
				root = st.top();
				st.pop();
				while (root) {
					lr = root;
					if (!hash[root]) {
						ans.push_back(root->val);
						cout << root->val << endl;
						hash[root] = 1;
					}
					root = root->left;
					lr->left = NULL;
					st.push(lr);
				}
				root = st.top();
				st.pop();
				if (root->right) {
					st.push(root->right);
				}
			}
		}
		return ans;
	}
};

 

迭代的力量!!!好强啊!

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值