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

本文关键字:不同的事 元素 foreach 最后一个 IEnumerable | 更新日期: 2023-09-27 18:35:23

我有一个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()像循环整个事情一样昂贵。

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

既然你提到了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"
    }
}

类似于 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);
    }
}

如果您想尽可能高效地执行此操作,除了有效地查看当前项目之外,别无选择,只能有效地查看"下一个"或"上一个"项目,因此您可以在获得该信息后推迟决定要做什么。例如,假设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);
}

看到它的实际效果

不是

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);

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

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