建立二叉树: 可以使用递归或者迭代的方式来建立二叉树。
递归方式建立二叉树:
示例代码如下:
struct TreeNode {
int val;
TreeNode* left;
TreeNode* right;
TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
};
TreeNode* buildTree(vector<int>& nums, int start, int end) {
if (start > end) {
return nullptr;
}
int mid = start + (end - start) / 2;
TreeNode* root = new TreeNode(nums[mid]);
root->left = buildTree(nums, start, mid - 1);
root->right = buildTree(nums, mid + 1, end);
return root;
}
TreeNode* buildBinaryTree(vector<int>& nums) {
return buildTree(nums, 0, nums.size() - 1);
}
迭代方式建立二叉树:
示例代码如下:
struct TreeNode {
int val;
TreeNode* left;
TreeNode* right;
TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
};
TreeNode* buildBinaryTree(vector<int>& nums) {
if (nums.empty()) {
return nullptr;
}
TreeNode* root = new TreeNode(nums[0]);
queue<TreeNode*> q;
q.push(root);
int i = 1;
while (i < nums.size()) {
TreeNode* node = q.front();
q.pop();
if (i < nums.size() && nums[i] != -1) {
node->left = new TreeNode(nums[i]);
q.push(node->left);
}
i++;
if (i < nums.size() && nums[i] != -1) {
node->right = new TreeNode(nums[i]);
q.push(node->right);
}
i++;
}
return root;
}
遍历二叉树: 常见的二叉树遍历方式有前序遍历、中序遍历和后序遍历。
前序遍历(根左右):
示例代码如下:
void preorderTraversal(TreeNode* root) {
if (root == nullptr) {
return;
}
// 访问根节点
cout << root->val << " ";
// 前序遍历左子树
preorderTraversal(root->left);
// 前序遍历右子树
preorderTraversal(root->right);
}
中序遍历(左根右):
示例代码如下:
void inorderTraversal(TreeNode* root) {
if (root == nullptr) {
return;
}
// 中序遍历左子树
inorderTraversal(root->left);
// 访问根节点
cout << root->val << " ";
// 中序遍历右子树
inorderTraversal(root->right);
}
后序遍历(左右根):
示例代码如下:
void postorderTraversal(TreeNode* root) {
if (root == nullptr) {
return;