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

时间:2012-05-24 10:32:10

标签: c# .net ienumerable control-flow

我有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)。 NB。如果我的IEnumerable恰好是长度1,我想要g(1)。

我的IEnumerable恰好有些蹩脚,让Count()和整个事情一样昂贵。

5 个答案:

答案 0 :(得分:21)

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

答案 1 :(得分: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);
    }
}

答案 2 :(得分:1)

如果你想尽可能高效地做到这一点,除了有效地不仅关注当前项目以及“下一个”或“前一个”项目,别无其他选择,因此你可以推迟决定在你之后做什么有那些信息。例如,假设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);
}

<强> See it in action

答案 3 :(得分:0)

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

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

答案 4 :(得分:0)

不是100%肯定我喜欢这个,但是你总是可以延迟使用item,直到你向IEnumerable数组移动一步,这样当你到达目的地时你没有使用过最后一个条目。 / p>

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

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