코딩성장스토리

백준 1764번:듣보잡 본문

백준 코딩

백준 1764번:듣보잡

까르르꿍꿍 2022. 1. 27. 17:31

https://www.acmicpc.net/problem/1764

 

1764번: 듣보잡

첫째 줄에 듣도 못한 사람의 수 N, 보도 못한 사람의 수 M이 주어진다. 이어서 둘째 줄부터 N개의 줄에 걸쳐 듣도 못한 사람의 이름과, N+2째 줄부터 보도 못한 사람의 이름이 순서대로 주어진다.

www.acmicpc.net

이 문제는 이진트리로 푸는 문제지만 그런 자료구조 보다는 set에 대해 더 공부 한것 같다. 컴파일 에러가 무진장 많이 떠서 무지하게 찾아봤다.

일단 set의 값은 고유하기 때문에 직접 바꾸는게 불가능하여 다시 직접 넣어 줘야한다. 

난 pair을 써서 문자열과 정수를 받았고 똑같은 문자열이 있으면 정수를 0에서 1로 바꾸었다.

그리고 정수가 1인 문자만 출력했다.

코드를 보자

#include <iostream>
#include <set>
#include <string>
using namespace std;
int main(){
    int n,m;
    string str;
    set<pair<string,int>>::iterator iter;
    cin>>n>>m;
    set<pair<string,int>> s;
    while(n--){
        cin>>str;
        s.insert(make_pair(str,0));
    }
    int cnt=0;
    while(m--){
        cin>>str;
        iter=s.find(pair<string,int>(str,0));
        if(iter!=s.end()){
            s.insert(iter,make_pair(str,1));       //set값은 고유 하기 때문에 다시 second값 1로 삽입 1이면 이미 있는 문장
            cnt++;                                  //갯수
        }
    }
    cout << cnt<<'\n';
    for(iter=s.begin();iter!=s.end();iter++){        //출력
        if((*iter).second==1){
            cout << (*iter).first<<'\n';
        }
    }
    
}