我有以下代码：
var e = someList.GetEnumerator();
var a = new List<Foo>();
var b = new List<Foo>();
while(e.MoveNext()) {
if(CheckCondition(e.Current)) {
b.Add(e.Current);
break;
}
a.Add(e.Current);
}
while(e.MoveNext())
b.Add(e.Current)
这看起来很丑陋基本上,遍历列表并将元素添加到一个列表中,直到某些条件启动,并将其余的添加到另一个列表.
有没有更好的方法,例如使用linq？ CheckCondition()是昂贵的,列表可能是巨大的,所以我宁愿不做任何迭代列表两次.
解决方法
这是一个解决方案,将枚举列表两次,但它不会第二次检查条件,所以它应该更快：
var a = someList.TakeWhile(x => !CheckCondition(x)).ToList(); var b = someList.Skip(a.Count).ToList();
如果someList实现IList< T>,则每个项目实际上仅被枚举一次,因此不会有任何惩罚.
我认为Skip针对IList< T>的情况进行了优化,但显然不是…但是您可以轻松实现使用此优化的自己的Skip方法(请参阅Jon Skeet’s article关于此)
如果有一个TakeUntil方法,它会更加优雅…我们可以轻松创建它：
public static IEnumerable<TSource> TakeUntil<TSource>(this IEnumerable<TSource> source,Func<TSource,bool> predicate)
{
foreach(var item in source)
{
if (predicate(item))
break;
yield return item;
}
}
var a = someList.TakeUntil(CheckCondition).ToList(); var b = someList.Skip(a.Count).ToList();
