替换c#树中的单个节点

时间:2017-07-10 12:32:50

标签: c# .net tree tree-traversal

我的应用程序中有一个树构造,其中包含用户的分层数据。

public class User {
     //a lot of properties here like Id, Name, Last Name, etc.
     public IEnumerable<User> Employees {get;set;}
}

现在我需要用新项替换此树中的某些记录。

var changedUsers = GetChangedUsers();
// Replace all changed user fields in original hierarchy 
// (only FirstName, LastName, etc. without touching Employees field)

有没有一种优雅的方法来实现这一目标?

修改

我试图用递归循环,但坚持更新记录。

private void ReplaceUserInHierarchy(User modifiedUser, List<User> users)
        {
            foreach (var user in users)
            {
                if (user.Id == modifiedUser.Id)
                {
                    //we should update here somehow
                    return;
                }

                ReplaceUserInHierarchy(modifiedUser, user.Employees);
            }
        }

1 个答案:

答案 0 :(得分:2)

只需更新与给定ID匹配的用户:

private void ReplaceUserInHierarchy(User modifiedUser, List<User> users)
{
    foreach (var user in users)
    {
        if (user.Id == modifiedUser.Id)
        {
            // update properties of user here
            user.FirstName = modifiedUser.FirstName;
            // etc
            return; // if user can be duplicated in hierarchy, then use continue here
        }

        // assume user cannot be in own employees hierarchy
        ReplaceUserInHierarchy(modifiedUser, user.Employees);
    }
}

改善优雅 - 您可以使用扩展层次结构的扩展方法。然后搜索您应该更新的用户:

var user = users.Flatten(u => u.Employees).FirstOrDefault(u => u.Id == modifiedUser.Id);
if (user != null) // you can throw if user not found
    user.FirstName = modifiedUser.FirstName; // etc        

展平可以

完成
public static IEnumerable<T> Flatten<T>(
   this IEnumerable<T> source,
   Func<T, IEnumerable<T>> selector)
{
   // null-check arguments
   foreach(var item in source)
   {
       yield return item;
       foreach(var child in Flatten(selector(item), selector))
           yield return child;
   }   
}

进一步改进 - 您可以将扁平化层次结构保存到字典中,并使用它来更新多个用户:

var usersById = users.Flatten(u => u.Employees).ToDictionary(u => u.Id);
foreach(var modifiedUser in modifiedUsers)
{
    User user;
    if (!usersById.TryGetValue(modifiedUser.Id, out user);)
       continue; // or throw

    user.FirstName = modifiedUser.FirstName; // etc  
}

进一步改进 - 您可以使用AutoMapper之类的库来自动进行映射。