将平面数据加载到树数据结构中

时间:2018-02-22 15:18:39

标签: c# recursion tree parent-child treenode

目前,我有一个Item类和一个表示从SQL Server 2012数据库查询中检索的平面数据的项目列表。这是Item类。

public class Item
{
    public string Parent { get; set; }
    public string Child { get; set; }
    public string Name { get; set; }
    public int Quantity { get; set; }
}

我需要将此平面数据转换为树结构,并能够在执行某些操作时遍历树。这是一些数据的视觉效果。我的树将始终有一个根节点。

Tree Data and Structure

我已经看过几篇关于SO的帖子,但我还在努力解决这个问题。这是我拥有的通用TreeNode类。

public class TreeNode<T>
{
    private readonly T _value;
    private readonly List<TreeNode<T>> _children = new List<TreeNode<T>>();

    public TreeNode(T value)
    {
        _value = value;
    }

    public TreeNode<T> this[int i]
    {
        get { return _children[i]; }
    }

    public TreeNode<T> Parent { get; private set; }

    public T Value { get { return _value; } }

    public ReadOnlyCollection<TreeNode<T>> Children
    {
        get { return _children.AsReadOnly(); }
    }

    public TreeNode<T> AddChild(T value)
    {
        var node = new TreeNode<T>(value) { Parent = this };
        _children.Add(node);
        return node;
    }

    public TreeNode<T>[] AddChildren(params T[] values)
    {
        return values.Select(AddChild).ToArray();
    }

    public bool RemoveChild(TreeNode<T> node)
    {
        return _children.Remove(node);
    }

    public void Traverse(Action<T> action)
    {
        action(Value);
        foreach (var child in _children)
            child.Traverse(action);
    }

    public IEnumerable<T> Flatten()
    {
        return new[] { Value }.Concat(_children.SelectMany(x => x.Flatten()));
    }
}

我无法将平面数据递归加载到树结构中。我也不了解如何利用Traverse方法中的Action在遍历时执行某些任务。为了填充树,我相信我需要从根开始。我这样得到它:

var root = new TreeNode<Item>(items.Single(i => i.Parent == null));

然后我可以像这样加载第一级:

root.AddChildren(items.Where(i => i.Parent.Equals(root.Value.Child)).ToArray());

我的两个问题如下:

  1. 我如何递归加载所有内容?
  2. 如何使用Traverse方法中的Action简单地通过具有适当缩进的项目名称写出树结构(或执行任何任务)?
  3. 提前致谢!!

2 个答案:

答案 0 :(得分:1)

以下是通常在课程中完成的方式

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Data;
using System.Xml;
using System.Xml.Linq;

namespace ConsoleApplication23
{
    class Program
    {
        static void Main(string[] args)
        {
            DataTable dt = new DataTable("_InvTrans");
            dt.Columns.Add("Parent", typeof(string));
            dt.Columns.Add("Child", typeof(string));
            dt.Columns.Add("Name", typeof(string));

            dt.Rows.Add(new object[] { "Null", "A", "Alpha" });
            dt.Rows.Add(new object[] { "A", "B", "Bravo" });
            dt.Rows.Add(new object[] { "A", "C", "Charlie" });
            dt.Rows.Add(new object[] { "B", "E", "Echo" });
            dt.Rows.Add(new object[] { "B", "F", "Foxtrot" });
            dt.Rows.Add(new object[] { "C", "W", "Whiskey" });
            dt.Rows.Add(new object[] { "C", "H", "Hotel" });

            new Node(dt);

        }

    }
    public class Node
    {
        public static Node root = new Node();
        static DataTable dt = null;

        public List<Node> Child { get; set; }
        public string Name { get; set; }
        public string id { get; set; }

        public Node(){}
        public Node(DataTable dt)
        {
            Node.dt = dt;
            root.id = "Null";
            Add(root);
        }

        public static void Add(Node parent)
        {
            foreach (DataRow row in Node.dt.AsEnumerable().Where(x => x.Field<string>("Parent") == parent.id))
            {
                Node child = new Node();
                if (parent.Child == null) parent.Child = new List<Node>();
                parent.Child.Add(child);
                child.Name = row.Field<string>("Name");
                child.id = row.Field<string>("Child");
                Add(child);
            }
        }
    }

}

答案 1 :(得分:1)

第一个问题:递归加载

您需要创建一个辅助函数(如果您使用C#7.0编写代码,则可以将其作为本地函数执行并删除son_of_your_mother = Child('Mother') 参数):

items

在获得root后按照你的例子调用它:

private static void AddDescendants(IReadOnlyCollection<Item> items, TreeNode<Item> node)
{
    var children = node.AddChildren(items.Where(i => i.Parent == node.Value.Child).ToArray());
    foreach (var child in children)
    {
        AddDescendants(items, child);
    }
}

第二个问题:遍历

您的var root = new TreeNode<Item>(items.Single(i => i.Parent == null)); AddDescendants(items, root); 函数执行pre-order traversal并且绝对不提供任何关于您所在树级别的信息,因此不能用于在输出中执行缩进。

如果您将实现更改为Traverse,请执行以下操作:

Action<TreeNode<T>>

然后您可以通过计算父母来计算缩进级别:

public void Traverse(Action<TreeNode<T>> action)
{
    action(this);
    foreach (var child in _children)
        child.Traverse(action);
}

这是一个包含输出的完整工作示例:https://ideone.com/faVOtd