Add size_t MapVector::erase(KeyT) similar to the one in std::map.

git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@219240 91177308-0d34-0410-b5e6-96231b3b80d8
This commit is contained in:
Kaelyn Takata 2014-10-07 21:15:51 +00:00
parent 9b303dcf12
commit 25cfb5cff5
2 changed files with 16 additions and 0 deletions

View File

@ -147,6 +147,17 @@ public:
return Next;
}
/// \brief Remove all elements with the key value Key.
///
/// Returns the number of elements removed.
size_type erase(const KeyT &Key) {
auto Iterator = find(Key);
if (Iterator == end())
return 0;
erase(Iterator);
return 1;
}
/// \brief Remove the elements that match the predicate.
///
/// Erase all elements that match \c Pred in a single pass. Takes linear

View File

@ -67,6 +67,11 @@ TEST(MapVectorTest, erase) {
ASSERT_EQ(MV.find(1), MV.end());
ASSERT_EQ(MV[3], 4);
ASSERT_EQ(MV[5], 6);
MV.erase(3);
ASSERT_EQ(MV.size(), 1u);
ASSERT_EQ(MV.find(3), MV.end());
ASSERT_EQ(MV[5], 6);
}
TEST(MapVectorTest, remove_if) {