代码改变世界

四行代码创建复杂（无限级）树

2011-12-09 20:38  鹤冲天  阅读(7531)  评论(16)  编辑  收藏  举报

最近两三天一直在做树方面的基础工作，碰巧今天在博客园看到一篇文章《C#中一种通用的树的生成方式》，粗略浏览下，感觉有不够强大。

对比而言，感觉自己的方式更好些，只需要四行代码就可以创建一颗复杂的无限级树。

在此分享一下，请大家先看两个运行截图：

image

image

下面，我们来看如何实现，先给出树节点的两个类：

树节点类

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
 
public class TreeNode {
    private List<TreeNode> _children;

    public TreeNode(string text, object value = null) {
        this.Text = text;
        this.Value = value;
        _children = new List<TreeNode>();
    }

    public string Text { get; private set; }
    public object Value { get; protected set; }
    public TreeNode Parent { get; set; }
    public IEnumerable<TreeNode> Children { get { return _children; } }

    public void Add(TreeNode childNode) {
        _children.Add(childNode);
        childNode.Parent = this;
    }
    public void Remove(TreeNode childNode) {
        _children.Remove(childNode);
    }
    public override string ToString() {
        return Text;
    }
}

一个比较标准的树节点，有父结点和多个子节点。Value 属性用来保存和树结点相关的对象的值。

再加上一个泛型版本的：

1
2
3
4
5
6
 
public class TreeNode<T> : TreeNode {
    public TreeNode(string text, T t)
        : base(text, t) {
    }
    public new T Value { get { return (T)base.Value; } }
}

算是一个强类型的树节点吧。

为了方便树节点的操作，我给 TreeNode 编写了一些扩展方法，如下给出本文要用的一个：

查找所有叶子节点的扩展方法

1
2
3
4
5
6
7
8
9
10
11
12
 
public static class TreeNodeExtensions {
    public static IEnumerable<TreeNode> GetLeafNodes(this TreeNode treeNode) {
        foreach (var child in treeNode.Children) {
            if (child.Children.Any()) {
                foreach (var descendant in GetLeafNodes(child))
                    yield return descendant;
            }
            else
                yield return child;
        }
    }
}

一个迭推调用。

最后到了重点：

TreeBuilder 相关类

TreeBuilder 类：

1
2
3
4
5
6
7
8
9
10
11
 
public static class TreeBuilder {
    public static BuildRootContext Build(string text) {
        var root = new TreeNode(text);
        return new BuildRootContext(root);
    }
    internal static TreeNode<T> BuildNode<T>(T t, Func<T, string> textSelect = null) {
        var text = textSelect != null ? textSelect(t) : Convert.ToString(t);
        var node = new TreeNode<T>(text, t);
        return node;
    }
}

BuildRootContext 类：

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
 
public class BuildRootContext {
    private TreeNode _tree;

    public BuildRootContext(TreeNode tree) {
        this._tree = tree;
    }

    public BuildChildrenContext<T> SetItems<T>(IEnumerable<T> items) {
        foreach (var item in items) {
            var node = TreeBuilder.BuildNode(item);
            _tree.Add(node);
        }
        return new BuildChildrenContext<T>(_tree);
    }
    public TreeNode Tree { get { return _tree; } }
}

BuildChildrenContext 类：

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
 
public class BuildChildrenContext<T> {
    private TreeNode _tree;

    public BuildChildrenContext(TreeNode tree) {
        this._tree = tree;
    }

    public TreeNode Tree { get { return _tree; } }

    public BuildChildrenContext<V> SetItems<V>(Func<T, IEnumerable<V>> itemSelector, Func<V, string> textSelect = null) {
        var leafNodes = _tree.GetLeafNodes().OfType<TreeNode<T>>();
        foreach (var leafNode in leafNodes) {
            foreach (var child in itemSelector(leafNode.Value)) {
                var node = TreeBuilder.BuildNode(child, textSelect);
                leafNode.Add(node);
            }
        }
        return new BuildChildrenContext<V>(_tree);
    }
    public BuildChildrenContext<T> SetRecursiveItems(Func<T, IEnumerable<T>> itemSelector,
        Func<T, string> textSelect = null) {
        var context = this;
        while (_tree.GetLeafNodes().OfType<TreeNode<T>>().Any(n => itemSelector(n.Value).Any()))
            context = context.SetItems<T>(itemSelector, textSelect);
        return context;
    }
}

通过这三个类即可达到文首图片中的效果，但实现可不是最佳方式，效率没太考虑，命名也不太好。

不过可以运行，先用上，慢慢改进吧！

简单使用说明

有限级树

像第一张图片中的产品树，只有两级：类别和产品，使用如下代码创建：

1
2
3
4
 
var tree = TreeBuilder.Build("产品")
    .SetItems(categories)
    .SetItems(category => products.Where(p => p.Category == category))
    .Tree;

第 1 行，构建根结点，传入一个字符串作为根结点的文本。

第 2 行，指定树的第一级，必须传入一个集合，上面传入的是 IEnumerable<Category>。

第 3 行，指定树的第二级，传入的是 Func<Category, IEnumerable<Product>>。

第 4 行，调用 Tree 属性，返回根结点，也是树了。

如果需要更多的级数，可将代码写在调用 Tree属性之前，也就是第 3 行和第 4 行之间。

增加一级，可以写成下面的样子（可没什么实际意义）：

1
2
3
4
5
 
var tree = TreeBuilder.Build("产品")
    .SetItems(categories)
    .SetItems(category => products.Where(p => p.Category == category))
    .SetItems(product=>product.Name)
    .Tree;

无限级树

第二张图片中的员工树是无限级的，因为员工的下属还可能有下属，是没有限制级数的。使用 SetRecursiveItems 方法构建：

1
2
3
4
 
var tree = TreeBuilder.Build("员工树")
    .SetItems(employees.Where(e => e.ReportsTo == null))
    .SetRecursiveItems(e => e.Subordinates)
    .Tree;

其代码它部分和有限级树相同。

还可以在员工树增加层次，之上增加部门，之后增加爱好等等，调用 SetItmes 即可。

部门无限然后员工无限，也是可以的。

总结

通过 TreeBuilder 可以让我们极其方便的创建很复杂的树，让我们把更多的精力放在业务逻辑上（而不是程序或界面逻辑）。

文中代码编写仓促，如有 Bug 请在回复中告知。

如果本文对你有帮助或启发，不妨推荐一下让更多的朋友看到。

 

在线演示：

你可以通过下面两个网址，看到通过本文的代码生成的树：

源码下载：TreeDemo.rar （15KB）