如何递归遍历嵌套的泛型集合？

Question

我想计算嵌套集合中类的数量，例如[[[1,2],[3,4,5]],[[1,2],[3,4,5]]]。这里引用的类是“int”，预期答案是 10。

我将此列表生成为：

        List<int> list1 = new List<int>(2);
        list1.Add(1);
        list1.Add(2);
        List<int> list2 = new List<int>(3);
        list2.Add(3);
        list2.Add(4);
        list2.Add(5);

        List<List<int>> listlist = new List<List<int>>(2);
        listlist.Add(list1);
        listlist.Add(list2);

        List<List<List<int>>> listlistlist = new List<List<List<int>>>(2);
        listlistlist.Add(listlist);
        listlistlist.Add(listlist);

当我在编程时，我更喜欢为泛型类编写这样的方法，我的代码是：

    public static int CountT<T>(ICollection<T> pTCol, int intLevel = 1)
    {
        int intCount = 0;
        if (intLevel > 0)
        {
            intLevel--;
            foreach (T pT in pTCol)
            {
                ICollection<T> subTCol = pT as ICollection<T>;   //PROBLEM: I got "null" for subTCol when the program was running
                intCount += CountT(subTCol, intLevel);
            }
        }
        else if (intLevel == 0)
        {
            intCount = pTCol.Count;
        }
        return intCount;
    }

我测试了上面的代码

        int intCount = CountT(listlistlist, 2);

那么问题来了

        ICollection<T> subTCol = pT as ICollection<T>;   //PROBLEM: I got "null" for subTCol when the program was running

我也试过代码：

    public static int CountT2<T, T2>(ICollection<T> pTCol, int intLevel = 1)
    {
        int intCount = 0;
        if (intLevel > 0)
        {
            intLevel--;
            foreach (T pT in pTCol)
            {
                ICollection<T2> subTCol = pT as ICollection<T2>;
                intCount += CountT2(subTCol, intLevel);   //PROBLEM: The type arguments for method cannot be inferred from the usage. Try specifying the type arguments explicitly. (I could not pass the compiling)
            }
        }
        else if (intLevel == 0)
        {
            intCount = pTCol.Count;
        }
        return intCount;
    }

编译不通过

        intCount += CountT2(subTCol, intLevel);   //PROBLEM: The type arguments for method cannot be inferred from the usage. Try specifying the type arguments explicitly. (I could not pass the compiling)

我该怎么做？

Answer 1

ICollection subTCol = pT as ICollection //问题：subTCol 得到“null”

这个为空的原因是pT 不是ICollection<T>，因为T 是int 而listlistlist 是List<List<List<T>>>。因此pT 是List<List<T>>，这就是为什么如果你尝试将它转换为ICollection<T> 你会得到null。

我想计算嵌套集合中的项目数。

您可以使用linq 的Sum 方法更轻松地做到这一点。如果您知道嵌套级别，例如：

Assert.AreEqual(10, listlistlist.Sum(x => x.Sum(y => y.Count)));

如果您不知道嵌套级别或者您想要更通用的方法，您可以创建一个扩展方法，例如：

public static class RecursiveSumExtension
{
    public static int RecursiveSum(this IEnumerable items)
    {
        if (null == items)
            return 0;

        var total = 0;

        foreach (var item in items)
        {
            if (item is IEnumerable)
                total += (item as IEnumerable).RecursiveSum();

            else
                total++;
        }

        return total;
    }
}

和测试：

 Assert.AreEqual(10, listlistlist.RecursiveSum());

列表生成

顺便说一句，您可以在创建集合时使用.net 的集合初始化语法使代码更具可读性：

        var listlist = new List<List<int>>
        {
            new List<int> {1, 2},
            new List<int> {3, 4, 5}
        };

        var listlistlist = new List<List<List<int>>>
        {
            listlist,
            listlist
        };