c++写一个树结构,支持添加节点,删除节点,获取节点,每个节点有个唯一id
时间: 2023-12-02 17:06:41 浏览: 174
结点A的度-C++版数据结构-张宏
下面是一个支持添加节点、删除节点、获取节点的树结构的 C++ 实现:
```c++
#include <iostream>
#include <unordered_map>
#include <vector>
using namespace std;
class TreeNode {
public:
int id;
int val;
vector<TreeNode*> children;
TreeNode(int id, int val) {
this->id = id;
this->val = val;
}
};
class Tree {
public:
unordered_map<int, TreeNode*> nodes;
Tree() {}
void addNode(int id, int val, int parentId) {
TreeNode* node = new TreeNode(id, val);
nodes[id] = node;
if (parentId != -1) {
nodes[parentId]->children.push_back(node);
}
}
void deleteNode(int id) {
TreeNode* node = nodes[id];
// 删除节点及其子节点
for (int i = 0; i < node->children.size(); i++) {
deleteNode(node->children[i]->id);
}
nodes.erase(id);
delete node;
}
TreeNode* getNode(int id) {
return nodes[id];
}
};
int main() {
// 创建一个树
Tree tree;
// 添加节点
tree.addNode(1, 1, -1);
tree.addNode(2, 2, 1);
tree.addNode(3, 3, 1);
tree.addNode(4, 4, 2);
tree.addNode(5, 5, 2);
tree.addNode(6, 6, 3);
// 遍历树
cout << "树的遍历结果:";
for (auto it = tree.nodes.begin(); it != tree.nodes.end(); it++) {
cout << it->second->val << " ";
}
cout << endl;
// 获取节点
TreeNode* node = tree.getNode(2);
cout << "节点 " << node->id << " 的值为 " << node->val << endl;
// 删除节点
tree.deleteNode(2);
cout << "删除节点 2 后,树的遍历结果:";
for (auto it = tree.nodes.begin(); it != tree.nodes.end(); it++) {
cout << it->second->val << " ";
}
cout << endl;
return 0;
}
```
这个树结构包含一个 unordered_map 来存储所有节点,每个节点有一个唯一的 id 和一个值,以及一个指向子节点的指针数组。在 Tree 类中,我们实现了添加节点、删除节点、获取节点的功能。在主函数中,我们创建了一个树并添加了多个节点,然后遍历树输出每个节点的值。接着,我们获取了一个节点并输出它的值,最后删除了一个节点并重新遍历树输出每个节点的值。
阅读全文