build_tree1.c

Author: realwujing

codes/c/chapter_divide_and_conquer/build_tree1.c

@@ -0,0 +1,43 @@
+/*
+ * 根据前序遍历和中序遍历构建二叉树的递归函数
+ *
+ * @param preorder 前序遍历数组指针
+ * @param inorderMap 中序遍历值到索引的映射数组
+ * @param i 当前子树在前序遍历数组中的根节点索引
+ * @param l 当前子树在中序遍历数组中的左边界
+ * @param r 当前子树在中序遍历数组中的右边界
+ * @param size 树的总节点数（用于边界检查）
+ * @return TreeNode* 构建好的子树根节点
+ */
+TreeNode *dfs(int *preorder, int *inorderMap, int i, int l, int r, int size) {
+	if (r - l < 0) {
+		return NULL;
+	}
+
+	TreeNode *root = (TreeNode *)malloc(sizeof(TreeNode));
+	root->val=preorder[i];
+	root->left = NULL;
+	root->right = NULL;
+	int m = inorderMap[preorder[i]];
+	root->left = dfs(preorder, inorderMap, i + 1, l, m - 1, size);
+	root->right = dfs(preorder, inorderMap, i + 1 + m -l, m + 1, r, size);
+	return root;
+}
+
+/*
+ * 根据前序遍历和中序遍历构建二叉树的主函数
+ *
+ * @param preorder 前序遍历数组指针
+ * @param preorderSize 前序遍历数组长度
+ * @param inorder 中序遍历数组指针
+ * @param inorderSize 中序遍历数组长度
+ * @return TreeNode* 构建好的二叉树根节点
+ */
+TreeNode *buildTree(int *preorder, int preorderSize, int *inorder, int inorderSize) {
+	int *inorderMap = (int *)malloc(sizeof(int) * MAX_SIZE);
+	for(int i = 0; i < inorderSize; i++) {
+		inorderMap[inorder[i]] = i;
+	}
+	TreeNode *root = dfs(preorder, inorderMap, 0, 0, inorderSize - 1, inorderSize);
+	free(inorderMap);
+}