利用两个遍历结果建立二叉链表存储结构,找到该二叉书中和为某一值的所有 路径
时间: 2024-03-31 16:36:48 浏览: 54
要实现这个功能,我们需要按照给定的两个遍历结果(比如先序遍历和中序遍历)建立二叉树,然后在二叉树中找到所有和为某一值的路径。这个问题可以通过递归实现。具体步骤如下:
1. 根据给定的先序遍历和中序遍历建立二叉树。这一步可以参考先前的问题:[如何根据一棵二叉树的先序遍历和中序遍历构建二叉树?](https://github.com/Yimuchen/ai-learn/blob/main/qa/how-to-build-a-tree-from-preorder-and-inorder-traversal.md)
2. 从根节点开始,对二叉树进行深度优先搜索,搜索所有可能的路径。
3. 在搜索过程中,记录当前路径上所有节点的值之和。如果当前路径的和等于给定的目标值,则将路径保存下来。
下面是用 C++ 实现的代码:
```cpp
#include <iostream>
#include <vector>
using namespace std;
struct TreeNode {
int val;
TreeNode* left;
TreeNode* right;
TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
};
void buildTree(TreeNode*& root, vector<int>& preorder, vector<int>& inorder, int preStart, int inStart, int inEnd) {
if (inStart > inEnd) return;
root = new TreeNode(preorder[preStart]);
int inIndex = inStart;
while (inIndex <= inEnd && inorder[inIndex] != preorder[preStart]) {
inIndex++;
}
buildTree(root->left, preorder, inorder, preStart + 1, inStart, inIndex - 1);
buildTree(root->right, preorder, inorder, preStart + inIndex - inStart + 1, inIndex + 1, inEnd);
}
void findPath(TreeNode* root, int target, vector<int>& path, int& sum, vector<vector<int>>& result) {
if (!root) return;
path.push_back(root->val);
sum += root->val;
if (!root->left && !root->right && sum == target) {
result.push_back(path);
}
findPath(root->left, target, path, sum, result);
findPath(root->right, target, path, sum, result);
sum -= root->val;
path.pop_back();
}
vector<vector<int>> pathSum(TreeNode* root, int target) {
vector<vector<int>> result;
vector<int> path;
int sum = 0;
findPath(root, target, path, sum, result);
return result;
}
int main() {
vector<int> preorder = {5, 4, 11, 7, 2, 8, 13, 4, 5, 1};
vector<int> inorder = {7, 11, 2, 4, 5, 13, 8, 5, 4, 1};
TreeNode* root;
buildTree(root, preorder, inorder, 0, 0, inorder.size() - 1);
vector<vector<int>> result = pathSum(root, 22);
for (auto path : result) {
for (int i = 0; i < path.size(); i++) {
cout << path[i];
if (i < path.size() - 1) cout << "->";
}
cout << endl;
}
return 0;
}
```
这里我们以先序遍历和中序遍历构建二叉树,并以二叉树根节点为起点进行搜索。在搜索过程中,使用 `path` 记录当前路径上的所有节点,使用 `sum` 记录当前路径上节点值之和。当搜索到叶子节点时,如果当前路径的和等于目标值,则将该路径保存到 `result` 中。最后输出所有符合条件的路径即可。
阅读全文