代码随想录算法训练营第15天|102. 二叉树的层序遍历,226.翻转二叉树,101.对称二叉树
102. 二叉树的层序遍历
题目链接:102. 二叉树的层序遍历,难度:中等
【代码实现】
//普通实现
/**
* 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:
vector<vector<int>> levelOrder(TreeNode* root) {
vector<vector<int>> result;
queue<TreeNode*> que;
if (root) {
que.push(root);
}
while (!que.empty()) {
vector<int> vec;
int size = que.size();
for (int i = 0; i < size; i++) {
TreeNode* tmp = que.front();
que.pop();
vec.push_back(tmp->val);
if (tmp->left) {
que.push(tmp->left);
}
if (tmp->right) {
que.push(tmp->right);
}
}
result.push_back(vec);
}
return result;
}
};
【解题思路】
把每一层的节点放进队列即可。
226.翻转二叉树
题目链接:226.翻转二叉树,难度:简单
【代码实现】
/**
* 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) {
stack<TreeNode*> st;
if (root) {
st.push(root);
}
while (!st.empty()) {
TreeNode* cur = st.top();
st.pop();
TreeNode* tmp = cur->left;
cur->left = cur->right;
cur->right = tmp;
if (cur->left) {
st.push(cur->left);
}
if (cur->right) {
st.push(cur->right);
}
}
return root;
}
};
//递归法
class Solution {
public:
TreeNode* invertTree(TreeNode* root) {
if (!root) {
return root;
}
swap(root->left, root->right);
invertTree(root->left);
invertTree(root->right);
return root;
}
};
【解题思路】
我最开始使用的是深度优先遍历,找到一个节点,交换左右节点即可。
101. 对称二叉树
题目链接:101. 对称二叉树,难度:简单
【代码实现】
/**
* 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:
bool compare(TreeNode* left, TreeNode* right) {
if (left == nullptr && right != nullptr) {
return false;
} else if (left != nullptr && right == nullptr) {
return false;
} else if (left == nullptr && right == nullptr) {
return true;
} else if (left->val != right->val) {
return false;
}
bool outside = compare(left->left, right->right);
bool inside = compare(left->right, right->left);
bool result = outside && inside;
return result;
}
bool isSymmetric(TreeNode* root) {
return compare(root->left, root->right);
}
};
【解题思路】
这里采用的后序遍历是这样的,通过比较后序遍历的递归写法可以看到,这里的写法也就是判断递归的返回条件和参数不同,重点理解的是通过后序遍历的方式进行节点的一些操作达到目的。