建立二叉樹: 可以使用遞歸或者迭代的方式來建立二叉樹。
遞歸方式建立二叉樹:
示例代碼如下:
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;