二叉樹|102.二叉樹層序遍歷、226.翻轉(zhuǎn)二叉樹泛范、101.對稱二叉樹
102.二叉樹層序遍歷
自己審題思路
使用隊列來維護每一層的節(jié)點
看完代碼隨想錄題解后的收獲
層序遍歷的遞歸法和延伸題目
代碼:
class Solution {
public:
vector<vector<int>> levelOrder(TreeNode* root) {
queue<TreeNode*> que;
if (root != NULL) que.push(root);
vector<vector<int>> result;
while (!que.empty()) {
int size = que.size();
vector<int> vec;
// 這里一定要使用固定大小size让虐,不要使用que.size(),因為que.size是不斷變化的
for (int i = 0; i < size; i++) {
TreeNode* node = que.front();
que.pop();
vec.push_back(node->val);
if (node->left) que.push(node->left);
if (node->right) que.push(node->right);
}
result.push_back(vec);
}
return result;
}
};
代碼(遞歸)
class Solution {
public:
void dfs(TreeNode* cur, vector<vector<int>>& vec, int depth) {
if (cur == nullptr) return;
if (vec.size() < depth) vec.push_back(vector<int>());
vec[depth - 1].push_back(cur->val);
dfs(cur->left, vec, depth + 1);
dfs(cur->right, vec, depth + 1);
}
vector<vector<int>> levelOrder(TreeNode* root) {
vector<vector<int>> ans;
dfs(root, ans, 1);
return ans;
}
};
遞歸法還是使用的
dfs
罢荡,但是給每層加了一個depth
標(biāo)志(從1開始)赡突,每次遍歷到depth
層時,向二維數(shù)組的depth-1
行(因為數(shù)組下標(biāo)從0開始)加入元素区赵。每當(dāng)二維數(shù)組的大胁宴帧(行數(shù))小于depth
時,表明遍歷到了新的一行惧笛,添加一行(添加一個空數(shù)組進去)从媚。
參考詳解
226.翻轉(zhuǎn)二叉樹
自己審題思路
遍歷過程中swap左右孩子
看完代碼隨想錄題解后的收獲
不能使用中序遍歷實現(xiàn)翻轉(zhuǎn)
代碼:
class Solution {
public:
void dfs(TreeNode* cur) {
if (cur == nullptr) return;
swap(cur->left, cur->right);
dfs(cur->left);
dfs(cur->right);
}
TreeNode* invertTree(TreeNode* root) {
dfs(root);
return root;
}
};
class Solution {
public:
vector<vector<int>> levelOrder(TreeNode* root) {
queue<TreeNode*> que;
if (root != NULL) que.push(root);
while (!que.empty()) {
int size = que.size();
for (int i = 0; i < size; i++) {
TreeNode* node = que.front();
que.pop();
swap(node->left, node->right);
if (node->left) que.push(node->left);
if (node->right) que.push(node->right);
}
}
return root;
}
};
參考詳解
101.對稱二叉樹
自己審題思路
判斷左右子樹是否對稱可以使用遞歸思想
看完代碼隨想錄題解后的收獲
具體的實現(xiàn)思路(各種方案的實現(xiàn)思路-----dfs,bfs患整,遞歸)
代碼:
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;
return compare(left->left, right->right) && compare(left->right, right->left);
}
bool isSymmetric(TreeNode* root) {
if (root == nullptr) return true;
return compare(root->left, root->right);
}
};
參考詳解
今日感悟
樹結(jié)構(gòu)的操作常用到遞歸思想