首页 > 解决方案 > TreeView 中的分层数据。在树中查找节点

问题描述

我有一个TreeView显示分层数据的地方。

模型

class Node : INotifyPropertyChanged
{
    public string Name { get; set; }
    public ObservableCollection<Node> Nodes { get; set; } = new ObservableCollection<Node>();
}

节点的存储:

class NodesStorage : ObservableCollection<Node>
{
    public NodesStorage()
    {
        var node = new Node { Name = "1" };
        node.Nodes.Add(new Node { Name = "1a" });
        node.Nodes.Add(new Node { Name = "1b" });
        node.Nodes.Add(new Node { Name = "1c" });

        var node1 = new Node { Name = "2" };
        node1.Nodes.Add(new Node { Name = "2a" });
        node1.Nodes.Add(new Node { Name = "2b" });
        node1.Nodes.Add(new Node { Name = "2c" });

        this.Add(node);
        this.Add(node1);
    }
}

视图模型

class MainWindowViewModel
{
    public MainWindowViewModel()
    {
        Nodes = new NodesStorage();
    }

    public ObservableCollection<Node> Nodes;
}

看法

<!--... Set DataContext ...-->

<TreeView ItemsSource="{Binding Path=Nodes}">
    <TreeView.ItemTemplate>
        <HierarchicalDataTemplate ItemsSource="{Binding Path=Nodes}">
            <TextBlock Text="{Binding Name}"/>
        </HierarchicalDataTemplate>
    </TreeView.ItemTemplate>
</TreeView>

<!--...-->

问题

现在我想创建返回的方法true,如果树包含具有以下名称的节点,并且false在另一种情况下:

public static bool IsTreeContains(Node node, string itemName)
{
    if (node.Name == itemName)
        return true;

    foreach (var n in node.Nodes)
    {
        return IsTreeContains(n, itemName);
    }

    // And what should I write here?
}

我想我可以在这里使用回溯。但这是一个非常糟糕的主意。
你怎么看?

标签: c#wpfmvvmtreetreeview

解决方案


我可以感觉到您正在树中进行“某种”深度优先搜索。

要逃避递归,您需要成功和失败两个条件,并且必须通过递归将结果传播给调用者。

public static bool IsTreeContains(Node node, string itemName)
{
    // If the current Node matches
    if (node.Name == itemName)
        return true;

    // End of the Tree or the Node Collection
    if (node.Nodes.Count == 0)
        return false;

    foreach (var n in node.Nodes)
    {
        if (IsTreeContains(n, itemName))
            return true;
    }

    return false;
}

我想我刚刚完成了你的解决方案,但是有很多方法可以通过算法解决这个问题。

希望这可以帮助。

编辑: 根据评论更正了正确的算法。

编辑 1: 更正了循环仅在第一个子节点上运行。谢谢@BionicCode。


推荐阅读