226. Invert Binary Tree的C++解法

最新推荐文章于 2020-09-04 17:24:55 发布

缪斯赤贫

最新推荐文章于 2020-09-04 17:24:55 发布

阅读量410

点赞数

分类专栏： LeetCode 文章标签： C++ leetcode

本文链接：https://blog.csdn.net/musechipin/article/details/69380689

版权

LeetCode 专栏收录该内容

158 篇文章 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 0;
		queue<TreeNode *> q;
		q.push(root);
		while (!q.empty())
		{
	           TreeNode *p = q.front();
	           q.pop();
			
	           TreeNode *temp=p->left;
	           p->left=p->right;
	           p->right=temp;
			    
	           if (p->left != NULL)
		     q.push(p->left);
		   if (p->right != NULL)
		     q.push(p->right);
		}
    return root;
	}
};

或者使用栈，算法基本是一个意思。

class Solution {
public:
TreeNode* invertTree(TreeNode* root) {
    std::stack<TreeNode*> stk;
    stk.push(root);
    
    while (!stk.empty()) {
        TreeNode* p = stk.top();
        stk.pop();
        if (p) {
            stk.push(p->left);
            stk.push(p->right);
            std::swap(p->left, p->right);
        }
    }
    return root;
}
};

没想到的是这个题也可以用递归。。仔细想想确实可以，因为子操作全部都是交换左右子树而已。

TreeNode* invertTree(TreeNode* root) {
    if (root) {
        invertTree(root->left);
        invertTree(root->right);
        std::swap(root->left, root->right);
    }
    return root;
}