c# - 在C#中管理人造树

我的任务是向现有代码库添加功能。该代码主要由刚离开我们团队的开发人员编写。考虑到这一点，我有一些限制：/。

开发人员以前已经编写了一些看起来像列表和树之间交叉的代码。在用户界面中，其为树形结构。但是，在代码中，它是一个列表。它有点奇怪。此处定义了两个主要类：

public class TreeBranch
{
  public Guid ID { get; set; }
}

public class TreeItem
{
  public Guid ID { get; set; }
  public Guid TreeBranchID { get; set; }
  public Guid? ParentTreeItemID { get; set; }

  public int Level { get; set; }

  public DateTime CreatedOn { get; set; }
  public DateTime? MarkedOn { get; set; }
}

因此，TreeBranch存储在数据库表中，而TreeItem存储在数据库表中。执行查询，以其MarkedOn值为NULL的特定TreeBranchID返回所有TreeItem元素。那就是处理列表：

List<TreeItem> treeItems = GetTreeBranchItems("someID");
treeItems.Sory(new TreeItemComparer());

我的TreeItemComparer类如下所示：

public class TreeItemComparer: IComparer<TreeItem>
{
  // allow us to look up parent Items by GUID
  IDictionary<Guid, TreeItem> itemLookup;

  public TreeItemComparer(IEnumerable<TreeItem> list)
  {
    itemLookup = list.ToDictionary(item => item.ID);
    foreach (var item in list)
      SetLevel(item);
  }

  public int SetLevel(TreeItem item)
  {
    if ((item.Level == 0) && (item.ParentTreeItemID != Guid.Empty))
    {
      if (itemLookup.ContainsKey(item.ParentTreeItemID))
        item.Level = 1 + SetLevel(itemLookup[item.ParentTreeItemID]);
    }
    else if (item.ParentTreeItemID == Guid.Empty)
      item.Level = 1;

    return item.Level;
  }

  public int Compare(TreeItem x, TreeItem y)
  {
    // see if x is a child of y
    while (x.Level > y.Level)
    {
      if ((x.ParentTreeItemID == y.ID) || (x.ParentTreeItemID == Guid.Empty))
        return 1;
      x = itemLookup[x.ParentTreeItemID];
    }

    // see if y is a child of x
    while (y.Level > x.Level)
    {
      if ((y.ParentTreeItemID == x.ID) || (y.ParentTreeItemID == Guid.Empty))
        return -1;
    }

    // x and y are not parent-child, so find common ancestor
    while (x.ParentTreeItemID != y.ParentTreeItemID)
    {
      if (x.ParentTreeItemID != Guid.Empty)
        x = itemLookup[x.ParentTreeItemID];
      if (y.ParentTreeItemID != Guid.Empty)
        y = itemLookup[y.ParentTreeItemID];
    }

    // compare createDate of children of common ancestor
    return x.CreatedOn.CompareTo(y.CreatedOn);
  }
}

此代码有效，但有一个例外。有时，我会折断树。例如，如果我有一个如下所示的结构：

Item 1
  Item 1-A  (Imagine this TreeItem has a MarkedOn value that is NOT null)
    Item 1-A-a
      Item 1-A-a-1
    Item 1-A-b
  Item 1-B

好吧，在上面的示例中，我应该只包含以下内容：

Item 1
  Item 1-B

但是，我越来越

Item 1
  Item 1-A-a
    Item 1-A-a-1
  Item 1-A-b
  Item 1-B

我的问题是，带有MarkedOn值的TreeItem及其所有“子项”不应属于集合的一部分。由于我们的环境，我无法接触数据库:(。同时，我不确定如何在代码中进行此操作。有人有什么想法吗？

谢谢！

最佳答案

从数据库取回结果后，可以过滤结果。

List<TreeItem> orphanedItems = new List<TreeItem>();
foreach (TreeItem item in results)
{
    if (item.ParentTreeItemID != Guid.Empty &&
        !results.Any(tree => tree.ID == item.ParentTreeItemID)
    {
        orphanedItems.Add(item);
    }
}
foreach (TreeItem orphan in orphanedItems)
    results.Remove(orphan);

关于c# - 在C#中管理人造树，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/15070627/