[프로그래머스/Level 2] 전화번호 목록 (C++)

programmers.co.kr/learn/courses/30/lessons/42577

코딩테스트 연습 - 전화번호 목록

 

풀이

 

문자열 벡터 phone_book에 존재하는 전화번호들의 각각의 번호를 key로 하여 그 개수를 value로 더해나갑니다. key, value형태의 데이터를 저장하기 위해서 unordered_map 자료구조를 이용하였습니다.

 

• 구조대 : 119
• 박준영 : 97 674 223
• 지영석 : 11 9552 4421

이와 같은 데이터 입력에서, phone_book의 첫번째 번호인 구조대의 번호를 다음과 같이 저장합니다.

 

prefix["1"]++;

prefix["11"]++;

prefix["119"]++;

 

다음으로 두번째 번호를 불러와서 다음과 같이 저장합니다.

 

prefix["9"]++;

prefix["97"]++;

....

prefix["97674223"]++

 

이런 식으로 모든 번호의 각 요소를 위와 같이 처리 해준 후, 번호들을 key값으로 하는 value 중 값이 2 이상인 경우가 존재하면 false를 리턴해주고, 아니면 true를 리턴하면 됩니다. 

 

전체 코드

 

#include <string>
#include <vector>
#include <unordered_map>
using namespace std;

bool solution(vector<string> phone_book) {
	unordered_map<string,int> prefix;
    
    for(string s : phone_book) {
    	string res="";
     	for(int i=0;i<s.size();i++) {
        	res+=s[i];
            prefix[res]++;
        }
    }
    
    for(string s : phone_book) {
    	if(prefix[s]>1) return false;
    }
    
    return true;
}