背景

private List<HuffmanTreeNode> CreateInitForest(string str)
{
    if (string.IsNullOrEmpty(str))
        throw new ArgumentNullException();

    List<HuffmanTreeNode> result = new List<HuffmanTreeNode>();
    char[] charArray = str.ToCharArray();
    List<IGrouping<char, char>> lst = charArray.GroupBy(a => a).ToList();

    foreach (IGrouping<char, char> g in lst)
    {
        char data = g.Key;
        int weight = g.ToList().Count;
        HuffmanTreeNode node = new HuffmanTreeNode(data, weight);
        result.Add(node);
    }
    return result;
}

该代码实现了统计字符串 str 中不同字符出现的频数，并把这个频数作为 Huffman 树结点的权值，字符作为数据，生成对应的 Huffman 树的叶子结点。

在这篇图文发布之后，很多同学问我，针对数据集合的扩展方法 GroupBy 如何使用，我在这里给大家总结一下。

技术分析

GroupBy 从字面意义上理解就是根据 By 指定的规则对数据进行分组，所谓的分组就是将一个“数据集合”划分成若干个“小区域”，然后针对若干个“小区域”进行数据处理。这里的 GroupBy 方法与 SQL 语句中的 GroupBy 功能是完全一样的。

下面，我通过例子给大家介绍GroupBy方法的应用。

Step1. 构造集合中的元素类型 Student。

public class Student
{
    public string Name { get; set; }
    public int Age { get; set; }
    public Student(string name, int age)
    {
        Name = name;
        Age = age;
    }
}

Step2. 构造一个Student的数据集合List<Student>。

List<Student> lst = new List<Student>
{
    new Student("赵一", 17),
    new Student("钱二", 16),
    new Student("张三", 19),
    new Student("李四", 18),
    new Student("王五", 16),
    new Student("刘六", 17),
    new Student("刘六", 19)
};

Step3. 将集合数据按照年龄分组，并把分组结果显示出来。

List<IGrouping<int, Student>> sameAge
    = lst.GroupBy(a => a.Age).ToList();

sameAge.ForEach(a =>
{
    Console.WriteLine("年龄为" + a.Key + "的人为：");
    
    a.ToList().ForEach(b =>
    {
        Console.WriteLine(b.Name);
    });
});

输出结果

Step4. 将集合数据按照年龄分组，并把分组中包含元素大于1的显示出来。

List<IGrouping<int, Student>> sameAge
    = lst.GroupBy(a => a.Age).Where(g => g.Count() > 1).ToList();

sameAge.ForEach(a =>
{
    Console.WriteLine("年龄为" + a.Key + "的人为：");
    a.ToList().ForEach(b =>
    {
        Console.WriteLine(b.Name);
    });
});