C ++迭代向量并删除匹配的字符串

时间:2015-02-11 20:24:41

标签: c++ iterator stdvector

我正在尝试编写字符串列表。用户可以添加到列表中或从列表中删除,也可以显示当前列表。

显示列表和添加到列表工作正常,但我无法通过迭代列表来找出匹配来删除用户的字符串。

我如何更改代码以解决此问题? 查看是否(答案== 3)

// InClassAssignment-FavouriteGameList.cpp : Defines the entry point for the console application.
//

#include "stdafx.h"
#include <string>
#include <cstdlib>
#include <iostream>
#include <vector>
#include <algorithm>
#include <ctime>
#include <cctype>

using namespace std;


int _tmain(int argc, _TCHAR* argv[])
{
    vector <string>  gameList;
    int answer = 0;
    bool cont = true;
    int size = 0;
    vector<string>::iterator iter;
    string addToList;
    string removeFromList;

    cout << "\tGame List" << endl;

    while (cont)
    {
        cout << "--------------------------------------------" << endl;
        cout << "\nWhat do you want to do?";
        cout << "\n1 - Display List";
        cout << "\n2 - Add to List";
        cout << "\n3 - Remove from List";
        cout << "\n4 - End program" << endl << "Selection: ";
        cin >> answer;

        cout << endl;

        if (answer == 1)
        {
            cout << "List: ";
            for (iter = gameList.begin(); iter != gameList.end(); ++iter)
            {
                if (iter != gameList.end() - 1)
                    cout << *iter << ", ";
                else
                    cout << *iter << endl;
            }
        }

        else if (answer == 2)
        {
            cout << "Type in a game to add: ";
            cin >> addToList;
            gameList.push_back(addToList);
            cout << "\nAdded (" << addToList << ") to your list." << endl;
        }

        else if (answer == 3)
        {
            //display list
            cout << "List: ";
            for (iter = gameList.begin(); iter != gameList.end(); ++iter)
            {
                if (iter != gameList.end() - 1)
                    cout << *iter << ", ";
                else
                    cout << *iter << "\n" << endl;
            }

            //ask which one to remove
            cout << "Which game should be removed?: ";
            cin >> removeFromList;

            //loop/iterate through the list to find a match and erase it
            for (iter = gameList.begin(); iter != gameList.end(); ++iter)
            {
                if ()
                    cout << "\nRemoved (" << removeFromList << ")" << endl;
                else
                    cout << "\nGame not found" << endl;
            }

        }

        else
        {
            cont = false;
        }

    }

    cout << "\nThanks for using the program!\n" << endl;

    return 0;
}

2 个答案:

答案 0 :(得分:5)

您可以使用std::find来获取与要删除的项目匹配的迭代器,然后调用vector::erase(iter)

auto iter = std::find(gameList.begin(), gameList.end(), removeFromList);
if (iter != gameList.end())
{
    gameList.erase(iter);
}

答案 1 :(得分:1)

看看remove。它会移动序列末尾的所有匹配元素,然后您可以截断它:

input.erase(
    remove(input.begin(), input.end(), s)
  , input.end());

编辑:汇总弗雷德评论的建议。