Как условно удалить элементы из коллекции .NET

Question

Я пытаюсь написать метод расширения в .NET, который будет работать с общей коллекцией и удалить все элементы из коллекции, соответствующие заданным критериям.

Это была моя первая попытка:

public static void RemoveWhere<T>(this ICollection<T> Coll, Func<T, bool> Criteria){ 
    foreach (T obj in Coll.Where(Criteria)) 
     Coll.Remove(obj); 
} 

Однако это бросает InvalidOperationException, «Коллекция была изменена, операция перечисления не может выполнить». Который имеет смысл, так что я сделал вторую попытку со вторым переменным сбором для хранения предметов, которые должны быть удалены и перебирать, что вместо того, чтобы:

public static void RemoveWhere<T>(this ICollection<T> Coll, Func<T, bool> Criteria){ 
    List<T> forRemoval = Coll.Where(Criteria).ToList(); 

    foreach (T obj in forRemoval) 
     Coll.Remove(obj); 
} 

Это проливает же исключение; Я не уверен, что я действительно понимаю, почему, поскольку «Свод» больше не перебирается, поэтому почему он не может быть изменен?

Если у кого-то есть предложения относительно того, как я могу заставить это работать, или лучший способ достичь того же, это было бы здорово.

Спасибо.

Answer 1

Для List<T>, это уже существует, как RemoveAll(Predicate<T>). Как таковой, я предлагаю вам сохранить имя (позволяющее ознакомиться со знакомством и приоритетом).

В принципе, вы не можете удалить во время итерации. Есть два основных варианта:

• использование индексатор на основе итерации (for) и удаление
• буфер элементы для удаления и удалить после foreach (как вы уже сделали)

Так возможно:

public static void RemoveAll<T>(this IList<T> list, Func<T, bool> predicate) { 
    for (int i = 0; i < list.Count; i++) { 
     if (predicate(list[i])) { 
      list.RemoveAt(i--); 
     } 
    } 
} 

Или в более общем случае для любого ICollection<T>:

public static void RemoveAll<T>(this ICollection<T> collection, Func<T, bool> predicate) { 
    T element; 

    for (int i = 0; i < collection.Count; i++) { 
     element = collection.ElementAt(i); 
     if (predicate(element)) { 
      collection.Remove(element); 
      i--; 
     } 
    } 
} 

Этот подход имеет то преимущество, что избегает большого количества дополнительных копий списка.

Answer 2

Я только что протестировал его, и ваш второй метод работает нормально (как и должно быть). Что-то еще должно идти не так, можете ли вы предоставить немного пример кода, который показывает проблему?

List<int> ints = new List<int> { 1, 2, 3, 4, 5, 6, 7, 8, 9, 10 }; 

ints.RemoveWhere(i => i > 5); 
foreach (int i in ints) 
{ 
    Console.WriteLine(i); 
} 

Получает:

1 
2 
3 
4 
5 

Answer 3

Как сказал Марк, List<T>.RemoveAll() это путь для списков.

Я удивлен, что ваша вторая версия не работает, хотя вы получили звонок по номеру ToList() после вызова Where(). Без вызова ToList() это будет иметь смысл (потому что это будет оценено лениво), но это должно быть хорошо, как есть. Не могли бы вы показать короткий, но полный пример этого сбоя?

EDIT: Что касается вашего комментария в вопросе, я до сих пор не могу получить его, чтобы он потерпел неудачу.Вот короткий, но полный примера, который работает:

using System; 
using System.Collections.Generic; 
using System.Linq; 

public class Staff 
{ 
    public int StaffId; 
} 

public static class Extensions 
{ 
    public static void RemoveWhere<T>(this ICollection<T> Coll, 
             Func<T, bool> Criteria) 
    { 
     List<T> forRemoval = Coll.Where(Criteria).ToList(); 

     foreach (T obj in forRemoval) 
     { 
      Coll.Remove(obj); 
     } 
    } 
} 

class Test 
{ 
    static void Main(string[] args) 
    { 
     List<Staff> mockStaff = new List<Staff> 
     { 
      new Staff { StaffId = 3 }, 
      new Staff { StaffId = 7 } 
     }; 

     Staff newStaff = new Staff{StaffId = 5}; 
     mockStaff.Add(newStaff); 
     mockStaff.RemoveWhere(s => s.StaffId == 5); 

     Console.WriteLine(mockStaff.Count); 
    } 
} 

Если бы вы могли предоставить подобную полного примера, который терпит неудачу, я уверен, что мы можем работать причину.

Answer 4

Я только попробовал ваш второй пример, и это, кажется, работает нормально:

Collection<int> col = new Collection<int>() { 1, 2, 3, 4, 5, 6, 7, 8, 9, 10 }; 
col.RemoveWhere(x => x % 2 != 0); 

foreach (var x in col) 
    Console.WriteLine(x); 
Console.ReadLine(); 

я не получаю исключение.

Answer 5

Другой вариант Marcs RemoveAll:

public static void RemoveAll<T>(this IList<T> list, Func<T, bool> predicate) 
{ 
    int count = list.Count; 
    for (int i = count-1; i > -1; i--) 
    { 
     if (predicate(list[i])) 
     { 
      list.RemoveAt(i); 
     } 
    } 
}