2015-02-11 2 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. erase действует для диапазона [begin(), end()] –

+0

Это сработало, спасибо, только что-то смутило, зачем использовать auto, а не писать вообще? – Will

+3

@ DieterLücking Фактически, для единственной формы аргумента 'erase' проверка необходима, потому что аргумент требуется разыменовать, а' end() 'is not. Он упоминается где-то в таблице для требований контейнера последовательности. Требование [указано в cppreference] (http://en.cppreference.com/w/cpp/container/vector/erase) также. – Praetorian

1

Посмотрите на remove. Он перемещает все соответствующие элементы в конце последовательности, вы можете затем обрезать его:

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

Edit: включены предложения от комментария Фреда.

+1

Я бы предпочел идиому [erase-remove] (https://en.wikipedia.org/wiki/Erase%E2%80%93remove_idiom). См. Http://stackoverflow.com/q/22860119/10077 –

+0

@FredLarson: Верно. Благодарю. –