QTreeWidget更新时如何保留选中状态？替代全量重建方案

嘿，这个问题我之前帮不少人处理过——每秒全量重建QTreeWidget不仅会丢选中状态，还会带来不必要的性能开销，界面甚至可能出现闪烁。下面给你两种可行的思路，其中第二种是最优方案：

思路一：临时方案——保存并恢复选中状态 #

如果暂时没法重构为增量更新，可以在全量重建前后，通过唯一标识保存和恢复选中项（注意不能直接存item指针，因为clear后旧item对象会被销毁）。

修改后的代码示例 #

void insertIntoTree(QTreeWidget *tree) {
    // 1. 保存当前选中项的唯一标识（推荐用业务唯一ID存到Qt::UserRole，比文本更可靠）
    QString selectedKey;
    QTreeWidgetItem *currentSelected = tree->currentItem();
    if (currentSelected) {
        // 若树是多层级，建议保存完整路径（比如父项ID+子项ID），避免重复文本匹配出错
        selectedKey = currentSelected->data(0, Qt::UserRole).toString();
        // 临时方案：如果没设置自定义ID，也可以用文本：selectedKey = currentSelected->text(0);
    }

    tree->clear();
    tree->setSelectionMode(QAbstractItemView::SingleSelection);
    
    // 原有的全量重建逻辑（注意：原代码tree.members()需替换为你的数据源长度）
    for (int i = 0; i < yourDataSource.size(); i++) {
        QTreeWidgetItem *parent = new QTreeWidgetItem(tree);
        parent->setText(0, yourDataSource[i].parentName);
        // 给父项绑定业务唯一ID
        parent->setData(0, Qt::UserRole, yourDataSource[i].parentId);
        
        QTreeWidgetItem *child = new QTreeWidgetItem(parent);
        child->setText(0, yourDataSource[i].childName);
        child->setData(0, Qt::UserRole, yourDataSource[i].childId);
    }

    // 2. 恢复选中状态
    if (!selectedKey.isEmpty()) {
        // 递归查找匹配标识的item
        QList<QTreeWidgetItem*> matchedItems = tree->findItems(selectedKey, Qt::MatchExactly | Qt::MatchRecursive, 0);
        if (!matchedItems.isEmpty()) {
            tree->setCurrentItem(matchedItems.first());
            // 可选：滚动到选中项，确保用户能看到
            tree->scrollToItem(matchedItems.first());
        }
    }
}

思路二：最优方案——增量更新节点而非全量重建 #

每秒全量重建是非常低效的，增量更新才是正确的方向：只对比新老数据，新增、删除或修改变化的节点，既保留选中状态，又提升性能、减少界面闪烁。

增量更新代码示例 #

假设你的数据源是包含唯一ID的结构体：

struct TreeData {
    QString parentId;
    QString parentName;
    QString childId;
    QString childName;
};

更新函数如下：

void updateTree(QTreeWidget *tree, const QList<TreeData>& newDataSource) {
    tree->setSelectionMode(QAbstractItemView::SingleSelection);

    // 1. 收集现有节点的ID映射（用Qt::UserRole存储的业务ID作为键）
    QHash<QString, QTreeWidgetItem*> existingItemMap;
    // 遍历顶层父项
    for (int i = 0; i < tree->topLevelItemCount(); ++i) {
        QTreeWidgetItem *parent = tree->topLevelItem(i);
        QString parentId = parent->data(0, Qt::UserRole).toString();
        existingItemMap[parentId] = parent;

        // 遍历子项
        for (int j = 0; j < parent->childCount(); ++j) {
            QTreeWidgetItem *child = parent->child(j);
            QString childId = child->data(0, Qt::UserRole).toString();
            existingItemMap[childId] = child;
        }
    }

    // 2. 遍历新数据，更新或添加节点
    for (const TreeData& data : newDataSource) {
        // 处理父项
        QTreeWidgetItem *parentItem = existingItemMap.value(data.parentId);
        if (!parentItem) {
            // 父项不存在，创建新项
            parentItem = new QTreeWidgetItem(tree);
            existingItemMap[data.parentId] = parentItem;
        }
        // 更新父项文本（如果有变化）
        if (parentItem->text(0) != data.parentName) {
            parentItem->setText(0, data.parentName);
        }
        parentItem->setData(0, Qt::UserRole, data.parentId);

        // 处理子项
        QTreeWidgetItem *childItem = existingItemMap.value(data.childId);
        if (!childItem) {
            // 子项不存在，创建新项
            childItem = new QTreeWidgetItem(parentItem);
            existingItemMap[data.childId] = childItem;
        }
        // 更新子项文本（如果有变化）
        if (childItem->text(0) != data.childName) {
            childItem->setText(0, data.childName);
        }
        childItem->setData(0, Qt::UserRole, data.childId);
    }

    // 3. 删除新数据中不存在的旧节点
    QSet<QString> newDataIds;
    for (const TreeData& data : newDataSource) {
        newDataIds.insert(data.parentId);
        newDataIds.insert(data.childId);
    }

    QList<QTreeWidgetItem*> itemsToDelete;
    for (auto it = existingItemMap.constBegin(); it != existingItemMap.constEnd(); ++it) {
        if (!newDataIds.contains(it.key())) {
            itemsToDelete.append(it.value());
        }
    }

    for (QTreeWidgetItem *item : itemsToDelete) {
        // 移除并删除节点（避免内存泄漏）
        if (item->parent()) {
            item->parent()->takeChild(item->parent()->indexOfChild(item));
        } else {
            tree->takeTopLevelItem(tree->indexOfTopLevelItem(item));
        }
        delete item;
    }
}

为什么增量更新更好？ #

• 选中状态会自动保留，因为节点对象没有被销毁，只是更新了内容
• 每秒调用的性能开销大幅降低，界面不会闪烁
• 代码扩展性更强，后续修改树结构更方便

补充说明 #

• 一定要用业务唯一ID（比如数据库ID、UUID）作为节点标识，不要依赖文本——文本可能重复，导致匹配出错
• 如果你的树是多层级（超过两层），可以把遍历逻辑改成递归实现，适配更复杂的结构

内容的提问来源于stack exchange，提问作者Marko Marinkovic