3108 : 정올 참여 학생 리스트 만들기 1

문제

정올 참여 학생 리스트 만들기 1

코드

#include <iostream>
#include <vector>
#include <map>
#include <algorithm>
using namespace std;
 
 
int N;
 
struct Data{
    int number;
    string name;
};
 
bool cmp(const Data &v1, const Data &v2){
    return v1.number<v2.number;
}
 
map<int,int>check;
 
int main(){
    cin>>N;
    vector<Data>v;
    for(int i=0; i<N; i++){
        char cmd; int number; string name;
        cin>>cmd>>number>>name;
        if(cmd=='I'){
            if(check[number]==0){
                check[number]=1;
                v.push_back({number,name});
            }
        }
        else if(cmd=='D'){
            if(check[number]==1){
                check[number]=0;
                for(int i=0; i<v.size(); i++){
                    if(v[i].number==number) v.erase(v.begin()+i);
                }
            }
        }
    }
    sort(v.begin(),v.end(),cmp);
    
    
    for(int i=0; i<5; i++){
        int find; cin>>find;
        cout<<v[find-1].number<<' '<<v[find-1].name<<'\n';
        
    }
    
    
}

풀이

머리 식히려고 선택한 문제였는데, 나름 까다로웠습니다.
map을 사용하여 현재Insert 된 상태인지의 여부를 확인해주었고, Delete입력 시 erase문을 사용하여 해당 데이터를 삭제하는 방법으로 풀이했습니다.

채점