数据结构二叉树的代码
时间: 2024-05-08 19:13:22 浏览: 10
二叉树是一种常见的数据结构,它由节点组成,每个最多有两个子节点,分别称为左子节点和右子。以下是一个简单的二叉树的代码示例:
```python
class TreeNode:
def __init__(self, value):
self.val = value
self.left = None
self.right = None
# 创建一个二叉树
root = TreeNode(1)
root.left = TreeNode(2)
root.right = TreeNode(3)
root.left.left = TreeNode(4)
root.left.right = TreeNode(5)
# 遍历二叉树(示例为前序遍历)
def preorder_traversal(node):
if node is not None:
print(node.val)
preorder_traversal(node.left)
preorder_traversal(node.right)
preorder_traversal(root)
```
这段代码定义了一个`TreeNode`类,每个节点包含一个值`val`以及左右子节点`left`和`right`。通过创建节点对象并设置它们的左右子节点,可以构建一个二叉树。代码中还提供了一个前序遍历函数`preorder_traversal`,用于按照前序遍历的顺序输出二叉树的节点值。
相关问题
c语言数据结构二叉树代码
下面是一个简单的二叉树的 C 语言代码实现,包括创建二叉树、插入节点、遍历等基本操作:
```c
#include <stdio.h>
#include <stdlib.h>
// 定义二叉树节点结构体
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
};
// 创建二叉树
struct TreeNode* createBinaryTree() {
struct TreeNode *root = (struct TreeNode*)malloc(sizeof(struct TreeNode));
printf("请输入节点的值(-1表示该节点为空):");
scanf("%d", &root->val);
if (root->val == -1) {
free(root);
return NULL;
}
printf("请输入%d的左子节点:", root->val);
root->left = createBinaryTree();
printf("请输入%d的右子节点:", root->val);
root->right = createBinaryTree();
return root;
}
// 插入节点
void insertNode(struct TreeNode* root, int val) {
if (root == NULL) {
root = (struct TreeNode*)malloc(sizeof(struct TreeNode));
root->val = val;
root->left = NULL;
root->right = NULL;
return;
}
if (val < root->val) {
if (root->left == NULL) {
struct TreeNode* node = (struct TreeNode*)malloc(sizeof(struct TreeNode));
node->val = val;
node->left = NULL;
node->right = NULL;
root->left = node;
} else {
insertNode(root->left, val);
}
} else if (val > root->val) {
if (root->right == NULL) {
struct TreeNode* node = (struct TreeNode*)malloc(sizeof(struct TreeNode));
node->val = val;
node->left = NULL;
node->right = NULL;
root->right = node;
} else {
insertNode(root->right, val);
}
}
}
// 先序遍历
void preOrderTraversal(struct TreeNode* root) {
if (root == NULL) {
return;
}
printf("%d ", root->val);
preOrderTraversal(root->left);
preOrderTraversal(root->right);
}
// 中序遍历
void inOrderTraversal(struct TreeNode* root) {
if (root == NULL) {
return;
}
inOrderTraversal(root->left);
printf("%d ", root->val);
inOrderTraversal(root->right);
}
// 后序遍历
void postOrderTraversal(struct TreeNode* root) {
if (root == NULL) {
return;
}
postOrderTraversal(root->left);
postOrderTraversal(root->right);
printf("%d ", root->val);
}
int main() {
struct TreeNode* root = createBinaryTree();
printf("先序遍历:");
preOrderTraversal(root);
printf("\n中序遍历:");
inOrderTraversal(root);
printf("\n后序遍历:");
postOrderTraversal(root);
printf("\n插入节点7后的中序遍历:");
insertNode(root, 7);
inOrderTraversal(root);
return 0;
}
```
这个二叉树代码实现包括创建二叉树、插入节点和遍历三个主要的操作。其中,节点的插入操作使用了递归的方法,遍历操作使用了三种不同的方法来实现。
数据结构二叉树实验代码
好的,我可以提供一份二叉树实验代码的示例,供你参考。以下是一个简单的二叉树实现,包括节点的结构体定义、创建、遍历、插入和删除操作等。
```
#include <stdio.h>
#include <stdlib.h>
// 二叉树节点结构体定义
typedef struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
} TreeNode;
// 创建二叉树
TreeNode *createTree() {
TreeNode *root = NULL;
int val;
printf("请输入二叉树的根节点值:");
scanf("%d", &val);
if (val == -1) { // 输入-1表示结束
return NULL;
}
root = (TreeNode *)malloc(sizeof(TreeNode));
root->val = val;
root->left = createTree(); // 递归创建左子树
root->right = createTree(); // 递归创建右子树
return root;
}
// 前序遍历二叉树
void preorderTraversal(TreeNode *root) {
if (root == NULL) {
return;
}
printf("%d ", root->val);
preorderTraversal(root->left);
preorderTraversal(root->right);
}
// 中序遍历二叉树
void inorderTraversal(TreeNode *root) {
if (root == NULL) {
return;
}
inorderTraversal(root->left);
printf("%d ", root->val);
inorderTraversal(root->right);
}
// 后序遍历二叉树
void postorderTraversal(TreeNode *root) {
if (root == NULL) {
return;
}
postorderTraversal(root->left);
postorderTraversal(root->right);
printf("%d ", root->val);
}
// 插入节点
void insertNode(TreeNode *root, int val) {
if (root == NULL) {
return;
}
if (val < root->val) { // 插入左子树
if (root->left == NULL) {
TreeNode *node = (TreeNode *)malloc(sizeof(TreeNode));
node->val = val;
node->left = NULL;
node->right = NULL;
root->left = node;
} else {
insertNode(root->left, val);
}
} else { // 插入右子树
if (root->right == NULL) {
TreeNode *node = (TreeNode *)malloc(sizeof(TreeNode));
node->val = val;
node->left = NULL;
node->right = NULL;
root->right = node;
} else {
insertNode(root->right, val);
}
}
}
// 删除节点
TreeNode *deleteNode(TreeNode *root, int val) {
if (root == NULL) {
return NULL;
}
if (val < root->val) { // 在左子树中删除
root->left = deleteNode(root->left, val);
} else if (val > root->val) { // 在右子树中删除
root->right = deleteNode(root->right, val);
} else { // 找到了要删除的节点
if (root->left == NULL) { // 只有右子树
TreeNode *temp = root->right;
free(root);
return temp;
} else if (root->right == NULL) { // 只有左子树
TreeNode *temp = root->left;
free(root);
return temp;
} else { // 左右子树都存在
TreeNode *temp = root->right;
while (temp->left != NULL) {
temp = temp->left;
}
root->val = temp->val;
root->right = deleteNode(root->right, temp->val);
}
}
return root;
}
// 主函数
int main() {
TreeNode *root = NULL;
int val;
root = createTree(); // 创建二叉树
printf("前序遍历:");
preorderTraversal(root); // 前序遍历
printf("\n中序遍历:");
inorderTraversal(root); // 中序遍历
printf("\n后序遍历:");
postorderTraversal(root); // 后序遍历
printf("\n请输入要插入的节点值:");
scanf("%d", &val);
insertNode(root, val); // 插入节点
printf("插入后的中序遍历:");
inorderTraversal(root); // 中序遍历
printf("\n请输入要删除的节点值:");
scanf("%d", &val);
root = deleteNode(root, val); // 删除节点
printf("删除后的中序遍历:");
inorderTraversal(root); // 中序遍历
return 0;
}
```
以上是一个简单的二叉树实现示例,你可以根据自己的需求进行修改和扩展。