13

我创建了一个函数来遍历字符串向量并删除长度为 3 或更短的任何字符串。这是使用 STL 算法库的一课。

我在函数工作时遇到了麻烦，但它不仅会删除长度为 3 或更短的字符串，而且还会将字符串“vector”附加到末尾。

输出应该是

This test vector

相反，它是

This test vector vector"

我该如何解决？

/*
* using remove_if and custom call back function, write RemoveShortWords 
* that accepts a vector<string> and removes all strings of length 3 or
* less from it. *shoot for 2 lines of code in functions.
*/

#include <iostream>
#include <string>
#include <algorithm>
#include <vector>
#include <iterator>
using namespace std;

bool StringLengthTest(string test) //test condition for remove_if algo.  
{
    return test.length() <= 3;
}

void RemoveShortWords(vector<string> &myVector)
{
    //erase anything in vector with length <= 3
    myVector.erase(remove_if(myVector.begin(),
                             myVector.end(),
                             StringLengthTest));
}

int main ()
{
    //add some strings to vector
    vector<string> myVector;
    myVector.push_back("This");
    myVector.push_back("is");
    myVector.push_back("a");
    myVector.push_back("test");
    myVector.push_back("vector");

    //print out contents of myVector (debugging)
    copy(myVector.begin(), myVector.end(), ostream_iterator<string>(cout," "));
    cout << endl; //flush the stream

    RemoveShortWords(myVector); //remove words with length <= 3

    //print out myVector (debugging)
    copy(myVector.begin(), myVector.end(), ostream_iterator<string>(cout," "));
    cout << endl;

    system("pause");
    return 0;
}
4

2 回答 2

31

如果将语句分开，最容易理解这一点：

auto iter(remove_if(myVector.begin(), myVector.end(), StringLengthTest));
myVector.erase(iter);

这两行与您的单行相同。现在应该清楚“错误”是什么。remove_if，首先工作。它遍历整个向量并将所有“选定”条目移动到“末尾”（更好地说：它将未选定的条目移动到前面）。运行后，它会将迭代器返回到剩余条目的“最后”位置，例如：

这个
测试
向量
test <- 迭代器指向这里
向量

然后你用一个迭代器运行擦除。这意味着您擦除了指向的单个元素 - 因此您擦除了“测试”元素。- 剩下的就是你所看到的。

要修复它，只需从 remove_if 返回的向量中擦除到 end()。：

myVector.erase(remove_if(myVector.begin(), myVector.end(), StringLengthTest), myVector.end()); //erase anything in vector with length <= 3
于 2012-01-29T14:32:21.210 回答
12

您应该使用擦除的两个参数形式：

myVector.erase(remove_if(myVector.begin(), myVector.end(), StringLengthTest),
               myVector.end());
于 2012-01-29T14:28:55.630 回答