主页 / 问题 / 881414
Accepted
Kir_Antipov
Asked:2020-09-15 01:08:23 +0000 UTC2020-09-15 01:08:23 +0000 UTC

通过掩码扩展 IEnumerable<T>

  • 772

说到Linq-one-liners）

一般来说，有这样一个任务：
有IEnumerable<string>一个包含一些行。还有int[]，它包含上述集合中应插入空行的位置

例子：

// IEnumerable<string> collection:
"Some"
"Strings"
"To"
"Test"
"Method"

// int[] mask:
1
2
4

// IEnumerable<string> result:
"Some"
""
""
"Strings"
""
"To"
"Test"
"Method"

美妙之Linq处在于在没有任何额外变量和显式循环的情况下转换集合。但是，如果不引入一个新的变量，我就无法解决这个问题，所以我想问你：哪个解决方案会更优雅？）

我的实现：

脑袋上：

int i = 0;
List<string> result = new List<string>();
foreach (string x in collection)
{
    while (mask.Contains(i)) 
    {
        result.Add(string.Empty);
        ++i;
    }
    result.Add(x);
    ++i;
}

相同，但有 SelectMany：

int i = 0;
IEnumerable<string> result = collection.SelectMany(x => {
    List<string> part = new List<string>();
    while (mask.Contains(i))
    {
        part.Add(string.Empty);
        ++i;
    }
    part.Add(x);
    ++i;
    return part; });

升级版：

这在这类任务中通常很重要，所以我澄清一下：它mask是一个有序数组

c#

2 个回答

  1. 实际上，@VladD给出了一个好主意：

    如果我们从掩码中减去序列 { 0, 1, 2, 3, ... }，我们会准确地得到原始序列的索引序列，其中需要使用 SelectMany 添加元素

    实际上，如果我们从每个第 0i个元素中减去mask N 0 i （ N 0是一个非负整数序列），我们将得到一个预先计算好的位置集合，其中必须插入空字符串

    简单地说，我们只需mask要从每个元素中减去它在数组中的位置

    收到需要插入空行的位置集合后，我们已经使用了相同的SelectMany

    最终代码：

    // Определим данные
IEnumerable<string> collection = new List<string> { "Some", "Strings", "To", "Test", "Method" };
IEnumerable<int> mask = new int[] { 1, 2, 4 };

// Просчитаем маску
mask = mask.Select((x, i) => x - i);

// Для каждого элемента выберем коллекцию, которая состоит из такого числа пустых строк, сколько раз индекс объекта 
// встречается в маске, а также из этого самого элемента
IEnumerable<string> result = collection.SelectMany((x, i) => new List<string>(Enumerable.Repeat(string.Empty, mask.Count(index => index == i))) { x });

    结果：

    // IEnumerable<string> result:
"Some"
""
""
"Strings"
""
"To"
"Test"
"Method"

    事实上，我们得到了我们所期望的）

    再次感谢您的好主意！

    • 1
  2. Best Answer

    与预处理相同的想法的另一个变体。

    让我们创建一个字典：

    var dict = mask.OrderBy(n => n)
               .Select((n, idx) => n - idx - 1)
               .GroupBy(n => n)
               .ToDictionary(g => g.Key, g => g.Count());

    有了这个，我们得到一个这样的查询：

    Enumerable.Repeat("", dict.TryGetValue(-1, out var k) ? k : 0).Concat(
    collection.SelectMany((s, idx) =>
        Enumerable.Repeat("", dict.TryGetValue(idx, out var t) ? t : 0)
                  .Prepend(s))
);

    当掩码的索引为 0 时，需要第一行。

    • 1

相关问题