IEnumerable foreach，对最后一个元素做一些不同的事情

Question

我有一个IEnumerable<T> .我想为集合的每个项目做一件事，除了最后一项，我想对它做其他事情。我怎样才能整齐地编码？在伪代码中

foreach (var item in collection)
{
    if ( final )
    {
        g(item)
    }
    else
    {
        f(item)
    }
}

因此，如果我的IEnumerable是Enumerable.Range(1,4)我会做f（1） f（2） f（3） g（4）。KBK如果我的 IEnumerable 恰好是长度 1，我想要 g（1）。

我的IEnumerable碰巧有点蹩脚，使Count()像循环整个事情一样昂贵。

Answer 1

既然你提到了IEnumerable[<T>]（不是IList[<T>]等），我们就不能依赖计数等：所以我很想展开foreach：

using(var iter = source.GetEnumerator()) {
    if(iter.MoveNext()) {
        T last = iter.Current;
        while(iter.MoveNext()) {
            // here, "last" is a non-final value; do something with "last"
            last = iter.Current;
        }
        // here, "last" is the FINAL one; do something else with "last"
    }
}

请注意，从技术上讲，上述内容仅对IEnuemerable<T>有效;对于非通用，您需要：

var iter = source.GetEnumerator();
using(iter as IDisposable) {
    if(iter.MoveNext()) {
        SomeType last = (SomeType) iter.Current;
        while(iter.MoveNext()) {
            // here, "last" is a non-final value; do something with "last"
            last = (SomeType) iter.Current;
        }
        // here, "last" is the FINAL one; do something else with "last"
    }
}

Answer 2

类似于 Marc 的答案，但您可以编写一个扩展方法来包装它。

public static class LastEnumerator
{
    public static IEnumerable<MetaEnumerableItem<T>> GetLastEnumerable<T>(this IEnumerable<T> blah)
    {
        bool isFirst = true;
        using (var enumerator = blah.GetEnumerator())
        {
            if (enumerator.MoveNext())
            {
                bool isLast;
                do
                {
                    var current = enumerator.Current;
                    isLast = !enumerator.MoveNext();
                    yield return new MetaEnumerableItem<T>
                        {
                            Value = current,
                            IsLast = isLast,
                            IsFirst = isFirst
                        };
                    isFirst = false;
                } while (!isLast);
            }
        }
    }
}
public class MetaEnumerableItem<T>
{
    public T Value { get; set; }
    public bool IsLast { get; set; }
    public bool IsFirst { get; set; }
}

然后这样称呼它：

foreach (var row in records.GetLastEnumerable())
{
    output(row.Value);
    if(row.IsLast)
    {
        outputLastStuff(row.Value);
    }
}

Answer 3

如果您想尽可能高效地执行此操作，除了有效地查看当前项目之外，别无选择，只能有效地查看"下一个"或"上一个"项目，因此您可以在获得该信息后推迟决定要做什么。例如，假设T是集合中的项类型：

if (collection.Any()) {
    var seenFirst = false;
    T prev = default(T);
    foreach (var current in collection) {
        if (seenFirst) Foo(prev);
        seenFirst = true;
        prev = current;
    }
    Bar(prev);
}

看到它的实际效果。

Answer 4

不是

100% 确定我喜欢这个，但你总是可以延迟项目的使用，直到你进入 IEnumerable 数组一步，这样当你走到最后时，你还没有使用最后一个条目。

它避免了对枚举器强制计数。

object item = null;
foreach (var a in items)
{
  // if item is set then we can use it.
  if (item != null)
  {
      // not final item
      f(item);
  }
  item = a;
}
// final item.
g(item);

Answer 5

我真的不推荐它，但我想你可以做这样的事情......

object m_item = notPartOfListFlag = new object();
foreach(var item in enumerator){
   if(m_item != notPartOfListFlag)
   {
      //do stuff to m_item;
   }
   m_item = item;
}
//do stuff to last item aka m_item;

但是我会尝试使用某种集合来公开列表中项目的位置，然后使用

if(collection.IndexOf(item) == collection.Count-1) do stuff