Сгладить массив объектов, которые могут содержать массивы

Question

У меня есть IEnumerable<object>, который может содержать или не содержать некоторые вложенные коллекции. Например, моя точка может выглядеть примерно так:

[ "foo", 2, [1, 2, 3, 4], "bar" ] 

И я хочу, чтобы сгладить его:

[ "foo", 2, 1, 2, 3, 4, "bar" ] 

Я думал SelectMany должен работать здесь, но не вполне может найти правильная комбинация. Я мог бы это сделать, но я думал, что должно быть более элегантное решение.

Answer 1

IEnumerable<object> source = new object[] { "test", 1, new[] { 1, 2, 3 }, "test" }; 

var result = source .SelectMany(x => x is Array ? ((IEnumerable)x).Cast<object>() : Enumerable.Repeat(x, 1)); 

Чтобы получить его работы с вложенными массивами сделать лямбда-рекурсивных:

IEnumerable<object> source = new object[] { "test", 1, new object[] { 1, 2, new [] { "nested", "nested2" } }, "test" }; 

Func<IEnumerable<object>, IEnumerable<object>> flatten = null; 
flatten = s => s.SelectMany(x => x is Array ? flatten(((IEnumerable)x).Cast<object>()) : Enumerable.Repeat(x, 1)); 

var result = flatten(source); 

Answer 2

Это немного противно, но работает .. если предположить только один уровень:

a.SelectMany(t => 
    (t is IEnumerable<object>) 
     ? (IEnumerable<object>)t 
     : new object[] {t}) 

Я подозреваю, самым «элегантным» решением является на самом деле написать расширение.

public static IEnumerable<T> Flatten<T>(this IEnumerable<T> @this) { 
    foreach (var item in @this) { 
     if (item is IEnumerable<T>) { 
      foreach (var subitem in Flatten((IEnumerable<T>)item)) { 
       yield return subitem; 
      } 
     } 
     else yield return item; 
    } 
} 

Answer 3

Чуть короче рекурсивный альтернатива:

IEnumerable<object> source = new object[] { "test", 1, new object[] { 1, 2, new [] { "nested", "nested2" } }, "test" }; 

Func<IEnumerable<object>, IEnumerable<object>> flatten = null; 
flatten = s => s == null ? null : s.SelectMany(x => 
    flatten(x as IEnumerable<object>) ?? new [] { x } 
); 
var result = flatten(source);