主页 / 问题 / 822137
Accepted
Андрей Туманов
Asked:2020-05-02 18:56:51 +0000 UTC2020-05-02 18:56:51 +0000 UTC

使用位掩码对字典进行 LINQ 查询

  • 772

请帮助解决问题。鉴于：

public enum GroupType   {   None    = 0x0000
                        ,   Name    = 0x0001    
                        ,   Ext     = 0x0002
                        ,   Type    = 0x0004
                        };

Dictionary<int, GroupType> d;
d = new Dictionary<int, GroupType> { {1, GroupType.Name}
                                   , {2, GroupType.Name | GroupType.Type}
                                   , {3, GroupType.Type}
                                   };

结果应该是

Dictionary<GroupType, List<int>> dRes;
dRes = { {GroupType.Name, {1, 2}}
       , {GroupType.Type, {2, 3}}
       };

也就是说，对于每个位掩码值，必须分配原始集中存在的一组键。

c#

2 个回答

  1. Best Answer

    引入 GroupType.None 位掩码“0”是为了准确选择没有所有键的元素。比较是基于平等的。为了不比较== 0，而是做== GroupType.None

    Enum.HasFlag对于空掩码，它始终返回 true，包括。您将需要编写一个辅助方法来实现这样的比较：

    private static bool HasFlag(GroupType value, GroupType flag)
{
    if(flag==GroupType.None) return ((int) value)==0;
    return value.HasFlag(flag);
}

    之后，您可以将所有枚举值选择到一个列表中：

    var types = Enum.GetValues(typeof(GroupType))
                .Cast<GroupType>()
                .ToList();

    并使用列表来获取字典和分组值：

    var dRes = d
            //для каждой записи в словаре выбираем все типы в паре с ключом
            .SelectMany(entry => types.Where(t => HasFlag(entry.Value, t)).Select(t => new {t, entry.Key}))
            //группируем по типу
            .GroupBy(pair => pair.t, pair => pair.Key)
            //переводим в словарь
            .ToDictionary(group => group.Key, group => group.ToList());

    或者，可以types从字典中采样并为每种类型选择适当的键。

    ideone 演示：https ://ideone.com/OMCtY9

    • 3

  2. 这就是您可以达到预期结果的方法

        var enumValues = Enum.GetValues(typeof(GroupType));
    Dictionary<GroupType, List<int>> result = new Dictionary<GroupType, List<int>>();
    foreach (var p in d)
    {
        uint flags = (uint)p.Value;
        foreach (var enumValue in enumValues)
        {
            if ((flags & (int)enumValue) != 0)
            {
                GroupType group = (GroupType)enumValue;
                if (result.ContainsKey(group))
                {
                    List<int> list = result[group];
                    list.Add(p.Key);
                }
                else
                {
                    List<int> list = new List<int>() { p.Key };
                    result.Add(group, list);

                }
            }
        }
    }

    本质：我们从字典中读取值。接下来，我们遍历枚举的所有元素GroupType并检查枚举标志是否存在于当前值中（为此，我们执行操作&，如果标志未设置，则与位运算将返回 0，否则!= 0）。如果枚举存在于现有值中，那么我们将一个元素添加到结果字典中。

    • 2

相关问题