Skip to content

Commit 3833665

Browse files
committed
Implement retain for vec_deque
1 parent 9c88f3b commit 3833665

File tree

2 files changed

+38
-0
lines changed

2 files changed

+38
-0
lines changed

Diff for: src/libcollections/vec_deque.rs

+30
Original file line numberDiff line numberDiff line change
@@ -1396,6 +1396,36 @@ impl<T> VecDeque<T> {
13961396
// naive impl
13971397
self.extend(other.drain());
13981398
}
1399+
1400+
/// Retains only the elements specified by the predicate.
1401+
///
1402+
/// In other words, remove all elements `e` such that `f(&e)` returns false.
1403+
/// This method operates in place and preserves the order of the retained
1404+
/// elements.
1405+
///
1406+
/// # Examples
1407+
///
1408+
/// ```
1409+
/// let mut buf = VecDeque::new();
1410+
/// buf.extend(1..5);
1411+
/// buf.retain(|&x| x%2 == 0);
1412+
/// assert_eq!(buf, [2, 4]);
1413+
/// ```
1414+
#[stable(feature = "vec_deque_retain", since = "1.1.0")]
1415+
pub fn retain<F>(&mut self, mut f: F) where F: FnMut(&T) -> bool {
1416+
let len = self.len();
1417+
let mut del = 0;
1418+
for i in 0..len {
1419+
if !f(&self[i]) {
1420+
del += 1;
1421+
} else if del > 0 {
1422+
self.swap(i-del, i);
1423+
}
1424+
}
1425+
if del > 0 {
1426+
self.truncate(len - del);
1427+
}
1428+
}
13991429
}
14001430

14011431
impl<T: Clone> VecDeque<T> {

Diff for: src/libcollectionstest/vec_deque.rs

+8
Original file line numberDiff line numberDiff line change
@@ -885,3 +885,11 @@ fn test_append() {
885885
assert_eq!(b.iter().cloned().collect::<Vec<_>>(), [1, 2, 3, 4, 5, 6]);
886886
assert_eq!(a.iter().cloned().collect::<Vec<_>>(), []);
887887
}
888+
889+
#[test]
890+
fn test_retain() {
891+
let mut buf = VecDeque::new();
892+
buf.extend(1..4);
893+
buf.retain(|&x| x % 2 == 0);
894+
assert_eq!(buf, [2, 4]);
895+
}

0 commit comments

Comments
 (0)