2015-02-11 106 views
0

我想编写一个字符串列表。用户可以添加到列表中或从列表中删除以及显示当前列表。C++遍历一个向量并删除匹配的字符串

显示列表和添加到列表工作正常,但我不知道如何通过遍历列表中找到匹配来删除用户的字符串。

我该如何更改我的代码来解决这个问题? 看,如果(答案== 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; 
} 

回答

5

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

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

注意:您不需要if。擦除是有效的范围[开始(),结束()] – 2015-02-11 20:31:51

+0

它的工作感谢,唯一的事情我很困惑它为什么使用自动而不是写它呢? – Will 2015-02-11 20:35:47

+3

@DieterLücking实际上,对于'erase'的单一参数形式来说,检查是必需的,因为参数需要可解引用,而'end()'不需要。它在序列容器要求表中的某处提到。该要求也列在cppreference上(http://en.cppreference.com/w/cpp/container/vector/erase)。 – Praetorian 2015-02-11 20:39:36

1

看看在remove。它移动在序列结束时,所有匹配的元素,你便可以截断它:

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

编辑:从弗雷德的意见纳入建议。

+1

我更喜欢[erase-remove](https://en.wikipedia.org/wiki/Erase%E2%80%93remove_idiom)成语。请参阅http://stackoverflow.com/q/22860119/10077 – 2015-02-11 20:59:57

+0

@FredLarson:对。谢谢。 – 2015-02-12 00:00:46