如何移除树形结构中不在目标路径上的节点？

问题背景 #

我们有如下定义的树形节点类：

public class Item
{
    public string ItemName { get; set; }
    public double Value { get; set; }
    public int Id { get; set; }
    public int ParentId { get; set; }
    public List<Item> Children { get; set; } = new();
}

需求：找到指定节点（如Item 11）后，修剪整个树形结构，仅保留从根节点到该目标节点的唯一路径——路径上的每个父节点仅保留路径中的子节点，其余非路径节点全部移除（最终效果为Item 1→Item 4→Item 11的链式结构）。

两种可行实现思路 #

思路1：回溯路径+原地修剪 #

从目标节点向上回溯到根，记录完整路径后，逐个修剪路径上节点的子节点，只保留路径中的下一个节点：

步骤：

1. 收集路径：从目标节点出发，通过ParentId向上查找父节点，直到根节点，将所有路径节点存入列表。
2. 调整路径顺序：反转列表，得到从根到目标节点的正向路径。
3. 修剪子节点：遍历正向路径，对每个父节点，仅保留路径中的下一个节点作为子节点，删除其他所有子节点。

代码实现：

// 假设已通过团队提供的方法找到目标节点
Item targetNode = FindNodeByItemName(allNodes, "Item 11");

// 构建节点Id到实例的字典，方便快速查找父节点
var nodeDict = allNodes.ToDictionary(item => item.Id);

// 回溯收集路径
var path = new List<Item>();
var current = targetNode;
while (current != null)
{
    path.Add(current);
    // 根节点的ParentId可设为0（或其他不存在的Id），此时current会变为null，终止循环
    nodeDict.TryGetValue(current.ParentId, out current);
}

// 反转路径，得到根→目标的顺序
path.Reverse();

// 修剪每个节点的子节点
for (int i = 0; i < path.Count - 1; i++)
{
    var parentNode = path[i];
    var childToKeep = path[i + 1];
    // 仅保留路径中的子节点
    parentNode.Children = parentNode.Children.Where(c => c.Id == childToKeep.Id).ToList();
}

// 修剪后的根节点为path[0]
var trimmedRoot = path[0];

思路2：生成路径后重建树形 #

先获取根到目标的路径，再基于路径重新构建仅保留该路径的树形结构：

步骤：

1. 收集路径：同思路1，先获取从根到目标的完整路径列表。
2. 重建树形：从目标节点向上遍历路径，为每个父节点清空原有子节点，仅添加路径中的下一个节点。

代码实现：

// 同样先获取目标节点和节点字典
Item targetNode = FindNodeByItemName(allNodes, "Item 11");
var nodeDict = allNodes.ToDictionary(item => item.Id);

// 收集并反转路径
var path = new List<Item>();
var current = targetNode;
while (current != null)
{
    path.Add(current);
    nodeDict.TryGetValue(current.ParentId, out current);
}
path.Reverse();

// 重建树形结构
for (int i = path.Count - 1; i > 0; i--)
{
    var childNode = path[i];
    var parentNode = path[i - 1];
    // 清空父节点原有子节点，仅保留路径中的子节点
    parentNode.Children.Clear();
    parentNode.Children.Add(childNode);
}

var trimmedRoot = path[0];

注意事项 #

• 确保根节点的ParentId是一个不存在的Id（如0），避免回溯时出现死循环。
• 如果存在多棵独立树，需先确认目标节点所属的根节点，避免错误遍历其他树的节点。
• 团队提供的FindNodeByItemName方法需保证能准确返回目标节点（若存在重名节点，需额外处理）。

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