代码随想录说用前序遍历和后序遍历方便,而中序遍历比较绕。
活用自定义函数使程序结构更为清晰
这里用的是前序遍历,根结点root指向left和root,交换root的左右指向
执行过程:
也就是交换下例的 2 ,7
使得交换后,2后面指向的结点跟着到右边去了,7后面指向的左右结点跟着到左边的结点去了。
(突然想到可以做剪枝,遇到叶子结点就直接回溯。)
再交换2 后面的左右结点
7交换后面的左右结点
cpp
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
TreeNode* invertTree(TreeNode* root) {
bitree(root);
return root;
}
void bitree(TreeNode *root)
{
if(root==NULL)
return ;
sweap1(root,root->left,root->right);
bitree(root->left);
bitree(root->right);
}
void sweap1(TreeNode *root,TreeNode *left,TreeNode *right)
{
TreeNode *t=new TreeNode;
t=right;
root->right=root->left;
root->left=t;
}
};