给你一棵二叉树的根节点 root
,翻转这棵二叉树,并返回其根节点。
思路:
- 确定递归函数的参数和返回值
参数就是要传入节点的指针,不需要其他参数了,通常此时定下来主要参数,如果在写递归的逻辑中发现还需要其他参数的时候,随时补充。
返回值的话其实也不需要,但是题目中给出的要返回root节点的指针,可以直接使用题目定义好的函数,所以就函数的返回类型为TreeNode*
。
TreeNode* invertTree(TreeNode* root)
2.确定终止条件
当前节点为空的时候,就返回
if (root == NULL) return root;
3.确定单层递归的逻辑
swap(root->left, root->right);
invertTree(root->left);
invertTree(root->right);
代码如下:
/**
* 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) {
// if(root == NULL) return root;
// swap(root->left,root->right);
// invertTree(root->left);
// invertTree(root->right);
// return root;
// }
// };
//非递归
class Solution {
public:
TreeNode* invertTree(TreeNode* root) {
if (root == NULL) return root;
stack<TreeNode*> st;
st.push(root);
while(!st.empty()) {
TreeNode* node = st.top(); // 中0
st.pop();
swap(node->left, node->right);
if(node->right) st.push(node->right); // 右
if(node->left) st.push(node->left); // 左
}
return root;
}
};