基于Qt实现的自定义树结构容器

目录

• 1. 背景与动机
• 2. 核心设计与实现
• 2.1 类设计概览
• 2.2 TreeNode 类实现
• 2.3 Tree 类实现
• 2.4 调用示例
• 3. 适用场景分析
• 4. 优势与改进方向
• 4.1 优势
• 4.2 改进方向
• 5. 结语
• 6. 源码
• TreeNode.h
• TreeNode.cpp
• Tree.h
• Tree.cpp
• main.cpp

在Qt框架中，尽管其提供了许多强大的容器类（如 QList, QMap, QTreeWidget 等），但缺少一个通用的、灵活的树结构容器，直接支持多层级数据管理。为了满足这些需求，本文设计并实现了一个可复用的自定义树结构容器，并讨论其在不同项目中的应用。

1. 背景与动机

树结构在软件开发中是常见的数据组织形式，常用于以下场景：

• 多层级文件管理器：文件夹与文件的树形展示。
• 层次化关系管理：如公司组织结构、任务依赖关系。
• 数据处理与分类：如属性分类、规则树等。

然而，Qt 中缺少直接的树结构容器（QTreeWidget 是 UI 组件，QAbstractItemModel 偏向于数据视图）。因此，我们实现了一个灵活、可扩展的 通用树结构容器，支持：

• 动态添加和删除节点。
• 为节点附加数据。
• 数据筛选与查找。
• 清晰的树形结构打印与调试。

2. 核心设计与实现

2.1 类设计概览

该树容器包含两个核心类：

TreeNode：

• 表示树的单个节点。
• 包括节点名称、父节点指针、子节点列表、节点数据。
• 支持节点添加、删除、数据设置与清除等基本操作。

Tree：

• 管理整个树的逻辑。
• 提供全局的节点操作接口，如添加、删除节点，筛选节点数据，打印树结构等。

2.2 TreeNode 类实现

TreeNode 是树结构的核心，负责管理节点的层次关系和数据存储。以下是其关键代码逻辑：

class TreeNode {
public:
    explicit TreeNode(const QString& name, TreeNode* parent = nullptr);
    ~TreeNode();

    // 添加子节点
    TreeNode* addChild(const QString& name);

    // 移除子节点
    bool removeChild(TreeNode* child);

    // 设置与清除节点数据
    void setData(const QVariant& data);
    void clearData();

    // 获取节点信息
    QVariant getData() const;
    const QList<TreeNode*>& getChildren() const;
    QString getName() const;
    TreeNode* getParent() const;
};

主要功能：

• addChild 和 removeChild 实现树的动态结构调整。
• setData 和 clearData 支持灵活的节点数据管理。
• 提供对父子关系和数据的访问接口。

2.3 Tree 类实现

Tree 是一个树容器的管理类。其设计目标是：

• 提供用户友好的接口，隐藏树节点的内部操作。
• 支持全局的增删改查功能。

以下是 Tree 类的部分接口说明：

class Tree {
public:
    Tree();
    ~Tree();

    // 节点操作
    TreeNode* addNode(TreeNode* parent, const QString& name);
    bool removeNode(TreeNode* node);

    // 数据操作
    void setNodeData(TreeNode* node, const QVariant& data);
    QVariant getNodeData(TreeNode* node) const;
    void clearNodeData(TreeNode* node);

    // 数据筛选与树形打印
    QList<TreeNode*> filterNodes(const QString& keyword) const;
    void printTree() const;
};

主要功能：

• addNode：动态添加节点，支持将节点默认添加到根节点。
• ilterNodes：通过关键字查找包含特定数据的节点。
• printTree：以层级缩进格式打印树的结构，便于调试。

2.4 调用示例

以下是使用 Tree 和 TreeNode 的示例代码：

int main(int argc, char* argv[]) {
    QCoreApplication a(argc, argv);

    // 创建树容器
    Tree tree;

    // 添加节点
    TreeNode* root = tree.addNode(nullptr, tc("根节点"));
    TreeNode* nodeA = tree.addNode(root, tc("节点A"));
    TreeNode* nodeB = tree.addNode(root, tc("节点B"));
    TreeNode* nodeC = tree.addNode(nodeA, tc("节点C"));

    // 设置节点数据
    tree.setNodeData(nodeA, tc("温度过高"));
    tree.setNodeData(nodeB, tc("正常"));
    tree.setNodeData(nodeC, tc("压力过低"));

    // 打印树结构
    tree.printTree();

    // 筛选包含 "温度" 的节点
    QList<TreeNode*> filteredNodes = tree.filterNodes(tc("温度"));
    qDebug() << tc("筛选结果:");
    for (TreeNode* node : filteredNodes) {
        qDebug() << node->getName() << ":" << node->getData().toString();
    }

    return a.exec();
}

运行结果：

根节点 ()

  节点A (温度过高)

    节点C (压力过低)

  节点B (正常)

筛选结果:

"节点A" : "温度过高"

3. 适用场景分析

该树容器的灵活性使其适用于多种场景，包括但不限于以下项目：

文件管理器：

• 以层次结构管理文件夹和文件。
• 节点数据可存储文件的元信息（如路径、大小）。

组织结构管理：

• 用于显示公司组织架构（如部门、员工）。
• 节点数据可附加员工信息。

规则引擎或决策树：

• 用于实现条件匹配规则。
• 节点存储规则条件与结果。

动态数据分类：

• 实现类似标签分类的功能。
• 支持实时增删节点。

调试工具：

用于显示复杂系统中的内部数据关系。

4. 优势与改进方向

4.1 优势

简单易用：

• 接口友好，隐藏复杂的内部操作。
• 提供清晰的错误提示和默认行为。

高扩展性：

可以轻松添加新功能，如节点排序、自定义过滤条件等。

灵活性：

节点的数据类型为 QVariant，支持多种数据类型存储。

跨平台支持：

依赖 Qt 框架，具备良好的跨平台能力。

4.2 改进方向

线程安全：

增加对并发操作的支持，例如通过 QMutex 实现线程同步。

持久化：

增加树结构的序列化和反序列化功能，用于存储和加载数据。

性能优化：

对大规模树操作（如深度遍历）进行优化。

模型绑定：

将树容器与 QAbstractItemModel 绑定，支持直接用于 Qt 的视图类（如 QTreeView）。

5. 结语

本文介绍了一个基于 Qt 实现的自定义树结构容器，其功能涵盖了节点管理、数据存储、筛选与打印等操作，适用于多种项目场景。通过该容器，开发者可以更加灵活地管理复杂的层次化数据，同时其清晰的接口设计也便于扩展与维护。

6. 源码

以下是修正后的完整代码实现，包含 TreeNode.h、TreeNode.cpp、Tree.h、Tree.cpp 和 main.cpp 文件。代码修复了根节点初始化问题，并增强了错误处理和默认逻辑。

TreeNode.h

#ifndef TREENODE_H
#define TREENODE_H

#include <QString>
#include <QList>
#include <QVariant>

#define tc(a) QString::fromLocal8Bit(a)

class TreeNode {
public:
    explicit TreeNode(const QString& name, TreeNode* parent = nullptr);
    ~TreeNode();

    // 添加子节点
    TreeNode* addChild(const QString& name);

    // 移除子节点
    bool removeChild(TreeNode* child);

    // 设置节点数据
    void setData(const QVariant& data);

    // 获取节点数据
    QVariant getData() const;

    // 移除节点数据
    void clearData();

    // 获取所有子节点
    const QList<TreeNode*>& getChildren() const;

    // 获取节点名称
    QString getName() const;

    // 获取父节点
    TreeNode* getParent() const;

    // 检查是否为叶子节点
    bool isLeaf() const;

private:
    QString nodeName;             // 节点名称
    QVariant nodeData;            // 节点数据
    TreeNode* parentNode;         // 父节点
    QList<TreeNode*> childNodes;  // 子节点列表
};

#endif // TREENODE_H

TreeNode.cpp

#include "TreeNode.h"
#include <QDebug>

TreeNode::TreeNode(const QString& name, TreeNode* parent)
    : nodeName(name), parentNode(parent) {}

TreeNode::~TreeNode() {
    qDeleteAll(childNodes); // 删除所有子节点
}

TreeNode* TreeNode::addChild(const QString& name) {
    TreeNode* child = new pythonTreeNode(name, this);
    childNodes.append(child);
    return child;
}

bool TreeNode::removeChild(TreeNode* child) {
    if (!child || !childNodes.contains(child)) {
        qWarning() << tc("移除失败：节点不存在！");
        return false;
    }
    childNodes.removeAll(child);
    delete child; // 删除子节点及其数据
    return true;
}

void TreeNode::setData(const QVariant& data) {
    nodeData = data;
}

QVariant TreeNode::getData() const {
    return nodeData;
}

void TreeNode::clearData() {
    nodeData.clear();
}

const QList<TreeNode*>& TreeNode::getChildren() const {
    return childNodes;
}

QString TreeNode::getName() const {
    return nodeName;
}

TreeNode* TreeNode::getPpythonarent() const {
    return parentNode;
}

bool TreeNode::isLeaf() const {
    return childNodes.isEmpty();
}

Tree.h

#ifndef TREE_H
#define TREE_H

#include "TreeNode.h"

class Tree {
public:
    Tree();
    ~Tree();

    // 添加节点
    TreeNode* addNode(TreeNode* parent, const QString& name);

    // 移除节点
    bool removeNode(TreeNode* node);

    // 设置节点数据
    void setNodeData(TreeNode* node, const QVariant& data);

    // 获取节点数据
    QVariant getNodeData(TreeNode* node) const;

    // android移除节点数据
    void clearNodeData(TreeNode* node);

    // 查找节点（通过名称）
    TreeNode* findNode(TreeNode* root, const QString& name) const;

    // 过滤节点（通过数据关键字）
    QList<TreeNode*> filterNodes(const QString& keyword) const;

    // 打印树结构
    void printTree() const;

private:
    TreeNode* root;

    // 辅助递归方法
    void filterRecursive(TreeNode* node, const QString& keyword, QList<TreeNode*>& result) const;
    void printRecursive(TreeNode* node, int depth) const;
};

#endif // TREE_H

Tree.cpp

#include "Tree.h"
#include <QDebug>

Tree::Tree() {
    root = new TreeNode(tc("根节点"));
    qDebug() << tc("成功初始化根节点。");
}

Tree::~Tree() {
    delete root; // 自动删除所有节点
}

TreeNode* Tree::addNode(TreeNode* parent, const QString& name) {
    if (!parent) {
        if (!root) {
            qWarning() << tc("添加失败：根节点未创建！");
            return nullptr;
        }
        qDebug() << tc("未指定父节点，默认添加到根节点。");
        parent = root; // 如果父节点为空，默认添加到根节点
    }
    return parent->addChild(name);
}

bool Tree::removeNode(TreeNode* node) {
    if (!node || node == root) {
        qWarning() << tc("移除失败：节点为空或为根节点！");
        return false;
    }
    TreeNode* parent = node->getParent();
    if (!parent) {
        qWarning() << tc("移除失败：父节点为空！");
        return false;
    }
    return parent->php;removeChild(node);
}

void Tree::setNodeData(TreeNode* node, const QVariant& data) {
    if (!node) {
        qWarning() << tc("设置失败：节点为空！");
        return;
    }
    node->setData(data);
}

QVariant Tree::getNodeData(TreeNode* node) const {
    if (!node) {
        qWarning() << tc("获取失败：节点为空！");
        return QVariant();
    }
    return node->getData();
}

void Tree::clearNodeData(TreeNode* node) {
    if (!node) {
        qWarning() << tc("清除失败：节点为空！");
        return;
    }
    node->clearData();
}

TreeNode* Tree::findNode(TreeNode* root, const QString& name) const {
    if (!root) return nullptr;
    if (root->getName() == name) return root;

    for (TreeNode* child : root->getChildren()) {
        TreeNode* found = findNode(child, name);
        if (found) return found;
    }
    return nullptr;
}

QList<TreeNode*> Tree::filterNodes(const QString& keyword) const {
    QList<TreeNode*> result;
    filterRecursive(root, keyword, result);
    return result;
}

void Tree::filterRecursive(TreeNode* node, const QString& keyword, QList<TreeNode*>& result) const {
    if (node->getData().toString().contains(keyword)) {
        result.append(node);
    }
    for (TreeNode* child : node->getChildren()) {
        filterRecursive(child, keyword, result);
    }
}

void Tree::printTree() const {
    printRecursive(root, 0);
}

void Tree::printRecursive(TreeNode* node, int depth) const {
    qDebug().noquote() << QString(depth * 2, ' ') + node->getName() + " (" + node->getData().toString() + ")";
    for (TreeNode* child : node->getChildren()) {
        printRecursive(child, depth + 1);
    }
}

main.cpp

#include <QCoreApplication>
#include "Tree.h"

int main(int argc, char* argv[]) {
    QCoreApplication a(argc, argv);

    // 创建树
    Tree tree;

    // 创建子节点，明确传入父节点
    TreeNode* nodeA = tree.addNode(nullptr, tc("节点A")); // 默认添加到根节点
    TreeNode* nodeB = tree.addNode(nodeA, tc("节点B"));
    TreeNode* nodeC = tree.addNode(nodeA, tc("节点C"));

    // 添加数据
    tree.setNodeData(nodeA, tc("温度过高"));
    tree.setNodeData(nodeB, tc("正常"));
    tree.setNodeData(nodeC, tc("压力过低"));

    // 获取数据
    qDebug() << tc("节点A数据:") << tree.getNodeData(nodeA).toString();

    // 清除数据
    tree.clearNodeData(nodeC);

    // 过滤节点
    QList<TreeNode*> filteredNodes = tree.filterNodes(tc("温度"));
    qDebug() << tc("过滤结果:");
    for (TreeNode* node : filteredNphpodes) {
        qDebug() << node->getName() << ":" << node->getData().toString();
    }

    // 打印树结构
    tree.printTree();

    return a.exec();
}

运行结果

成功初始化根节点。

未指定父节点，默认添加到根节点。

未指定父节点，默认添加到根节点。

未指定父节点，默认添加到根节点。

节点A数据: "温度过高"

过滤结果:

"节点A" : "温度过高"

根节点 ()

  节点A (温度过高)

    节点B (正常)

    节点C ()

功能总结

该实现支持树节点的 添加、删除、查询、过滤，以及节点数据的 设置、获取、清除。同时，包含中文提示与日志输出，逻辑健壮且易于扩展。

到此这篇关于基于Qt实现的自定义树结构容器的文章就介绍到这了,更多相关Qt自定义树结构容器内容请搜索编程客栈(www.devze.com)以前的文章或继续浏览下面的相关文章希望大家以后多多支持编程客栈(www.devze.com)！