当前位置: 代码迷 >> 综合 >> leetcode 226. Invert Binary Tree(倒置一个二叉树,递归,非递归)
  详细解决方案

leetcode 226. Invert Binary Tree(倒置一个二叉树,递归,非递归)

热度:13   发布时间:2023-11-17 01:21:00.0

题目要求

倒置一个二叉树,可以尝试递归和非递归方法。

输入示例

Input:                                                4/   \2     7/ \   / \
1   3 6   9Output:4/   \7     2/ \   / \
9   6 3   1

解题思路

通过输入的实例,我们可以发现,实际上倒置一个二叉树就是将树中每一个节点的左右孩子进行交换。

1、递归
因此,我们可以对每个节点的左右孩子进行记录,然后互换,递归的对每个节点进行执行即可,根据这个思想可以写出递归的主要代码:

/*** 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:TreeNode* invertTree(TreeNode* root) {
    if(root == NULL) return NULL;TreeNode* left = invertTree(root->left); // 记录左孩子TreeNode* right = invertTree(root->right); // 记录右孩子// 进行交换操作root->left = right;root->right = left;return root;}
};

2、非递归
非递归的实现需要我们利用队列的特性(先进先出FIFO)来完成操作。具体执行的时候是把根节点放入队列,然后读取队列的首元素(除了叶子节点外就是双亲节点),然后交换这个元素的左右孩子,随后将交换后的左右孩子放入队列重复上面的操作,直到队列里面没有元素为止。

/*** 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:TreeNode* invertTree(TreeNode* root) {
    if(root == NULL) return NULL;queue<TreeNode*> q;q.push(root); // 根节点入队 操作开始while(!queue.empty()) // 队列中有元素的时候 重复以下操作{
    TreeNode* cur = q.front(); q.pop();swap(cur->left,cur->right); // 交换双亲节点的左右孩子if(cur->left != NULL) q.push(cur->left); // 交换后的左孩子入队if(cur->right != NULL) q.push(cur->right); // 交换后的右孩子入队}return root;     }
};

总结

正如以前说过的,在解决树类的大多数问题上,递归能迅速的给我们提供一个简单的解题思路,所以可以多用递归的思想去思考这类问题,特别的在进行二叉树的倒置,交换等操作的时候,队列是我们常用的辅助手段(排序是用栈),所以对队列不熟悉的朋友,要多多练习,记住我们常用的操作,和STL中的一些技巧。

入列:queue.push()
出列:queue.pop() 也是要率先指到队首元素(和栈一样),并不会返回值
队首元素:queue.front(),即最先被压入队列的元素
队尾元素:queue.back(),即最后被压入队列的元素
判断队列为空:queue.empty()

相似题目:
解答:leetcode 654. Maximum Binary Tree(构建最大二叉树)

原题链接:https://leetcode.com/problems/invert-binary-tree/

  相关解决方案