代码随想录说用前序遍历和后序遍历方便,而中序遍历比较绕。
活用自定义函数使程序结构更为清晰
这里用的是前序遍历,根结点root指向left和root,交换root的左右指向
执行过程:
也就是交换下例的 2 ,7
使得交换后,2后面指向的结点跟着到右边去了,7后面指向的左右结点跟着到左边的结点去了。
(突然想到可以做剪枝,遇到叶子结点就直接回溯。)
再交换2 后面的左右结点
7交换后面的左右结点
/*** 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;}};