生成一段二叉树代码
时间: 2023-07-10 08:07:10 浏览: 87
好的,以下是一个简单的二叉树代码示例:
```python
class TreeNode:
def __init__(self, val=0, left=None, right=None):
self.val = val
self.left = left
self.right = right
def build_tree(arr):
if not arr:
return None
mid = len(arr) // 2
root = TreeNode(arr[mid])
root.left = build_tree(arr[:mid])
root.right = build_tree(arr[mid+1:])
return root
```
这段代码定义了一个二叉树节点类 `TreeNode` 和一个构建二叉树的函数 `build_tree`。在 `build_tree` 函数中,我们首先判断数组是否为空,如果为空则返回空节点。否则,我们找到当前数组的中间位置,将其作为根节点,然后递归构建左子树和右子树,最后返回根节点。
相关问题
生成一段C语言二叉树生成代码
好的,以下是一段简单的 C 语言代码,用于生成二叉树:
```
#include <stdio.h>
#include <stdlib.h>
typedef struct node {
int data;
struct node *left;
struct node *right;
} Node;
Node* createNode(int data) {
Node* newNode = (Node*)malloc(sizeof(Node));
newNode->data = data;
newNode->left = NULL;
newNode->right = NULL;
return newNode;
}
Node* insert(Node* root, int data) {
if (root == NULL) {
return createNode(data);
}
if (data < root->data) {
root->left = insert(root->left, data);
} else {
root->right = insert(root->right, data);
}
return root;
}
void inorderTraversal(Node* root) {
if (root != NULL) {
inorderTraversal(root->left);
printf("%d ", root->data);
inorderTraversal(root->right);
}
}
int main() {
Node* root = NULL;
int n, data;
printf("Enter the number of nodes in the binary tree: ");
scanf("%d", &n);
printf("Enter the data for the nodes:\n");
for (int i = 0; i < n; i++) {
scanf("%d", &data);
root = insert(root, data);
}
printf("Inorder traversal of the binary tree: ");
inorderTraversal(root);
printf("\n");
return 0;
}
```
这个程序包含了二叉树节点的定义,创建节点和插入节点的函数,以及中序遍历二叉树的函数。在主函数中,用户可以输入二叉树的节点数和节点的数据,程序会自动创建二叉树并输出中序遍历结果。
生成一段递归生成二叉树的C语言代码
以下是一段递归生成二叉树的C语言代码:
```c
#include <stdio.h>
#include <stdlib.h>
struct Node {
int data;
struct Node* left;
struct Node* right;
};
struct Node* newNode(int data) {
struct Node* node = (struct Node*)malloc(sizeof(struct Node));
node->data = data;
node->left = NULL;
node->right = NULL;
return node;
}
struct Node* insert(struct Node* node, int data) {
if (node == NULL) {
return newNode(data);
}
if (data <= node->data) {
node->left = insert(node->left, data);
} else {
node->right = insert(node->right, data);
}
return node;
}
void inorderTraversal(struct Node* node) {
if (node == NULL) {
return;
}
inorderTraversal(node->left);
printf("%d ", node->data);
inorderTraversal(node->right);
}
int main() {
struct Node* root = NULL;
int arr[] = { 5, 3, 7, 1, 9, 4, 2, 6, 8 };
int n = sizeof(arr) / sizeof(arr[0]);
for (int i = 0; i < n; i++) {
root = insert(root, arr[i]);
}
inorderTraversal(root);
return 0;
}
```
在上面的代码中,我们定义了一个二叉树的节点结构体 `Node`,其中包含了节点的数据、左子树和右子树。我们使用 `newNode()` 函数创建一个新的节点,使用 `insert()` 函数插入一个新的数据到二叉树中,使用 `inorderTraversal()` 函数对二叉树进行中序遍历输出。
在 `main()` 函数中,我们定义了一个数组 `arr`,然后使用 `for` 循环将数组中的元素逐个插入到二叉树中。最后输出二叉树的中序遍历结果。
阅读全文