更新:
入れ子のレベル(深さ)に興味がある人向け。明示的な列挙スタックの実装の良い点の1つは、いつでも(特に要素を生成するとき)stack.Count
、現在の処理深度を表すことです。したがって、これを考慮してC#7.0値タプルを利用すると、メソッド宣言を次のように変更できます。
public static IEnumerable<(T Item, int Level)> ExpandWithLevel<T>(
this IEnumerable<T> source, Func<T, IEnumerable<T>> elementSelector)
とyield
ステートメント:
yield return (item, stack.Count);
次にSelect
、上記にsimple を適用して元のメソッドを実装できます。
public static IEnumerable<T> Expand<T>(
this IEnumerable<T> source, Func<T, IEnumerable<T>> elementSelector) =>
source.ExpandWithLevel(elementSelector).Select(e => e.Item);
元の:
驚くべきことに、誰も(Ericでさえ)は、再帰的なプレオーダーDFTの「自然な」反復ポートを示していなかったため、ここに示します。
public static IEnumerable<T> Expand<T>(
this IEnumerable<T> source, Func<T, IEnumerable<T>> elementSelector)
{
var stack = new Stack<IEnumerator<T>>();
var e = source.GetEnumerator();
try
{
while (true)
{
while (e.MoveNext())
{
var item = e.Current;
yield return item;
var elements = elementSelector(item);
if (elements == null) continue;
stack.Push(e);
e = elements.GetEnumerator();
}
if (stack.Count == 0) break;
e.Dispose();
e = stack.Pop();
}
}
finally
{
e.Dispose();
while (stack.Count != 0) stack.Pop().Dispose();
}
}