[프로그래머스] Hash(2) 전화번호 목록

문제 설명

전화번호부에 적힌 전화번호 중, 한 번호가 다른 번호의 접두어인 경우가 있는지 확인하려 합니다.
전화번호가 다음과 같을 경우, 구조대 전화번호는 영석이의 전화번호의 접두사입니다.

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

전화번호부에 적힌 전화번호를 담은 배열 phone_book 이 solution 함수의 매개변수로 주어질 때, 어떤 번호가 다른 번호의 접두어인 경우가 있으면 false를 그렇지 않으면 true를 return 하도록 solution 함수를 작성해주세요.

 

제한 사항

• phone_book의 길이는 1 이상 1,000,000 이하입니다.
• 각 전화번호의 길이는 1 이상 20 이하입니다.

입출력 예

phone_book	return
["119", "97674223", "1195524421"]	false
["123", "456", "789"]	true
["12", "123", "1235" , "567", "88"]	false

 

풀이

#include <string>
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;

bool solution(vector<string> phone_book) {
    bool answer = true;
    int cnt = 0;
    
    sort(phone_book.begin(), phone_book.end()); // phone_book 정렬
    
    for (int i = 1; i < phone_book.size(); i++) {
        for (int j = 0; j < phone_book[i-1].length(); j++) {
            
            if (phone_book[i-1][j] != phone_book[i][j]) break; // [i-1]과 [i] 단어 비교
            else cnt++; // 일치 확인을 위한 단어 길이 count
        }
        
        if (cnt == phone_book[i-1].length()) return false;  
    }
    
    return answer;
}

 

 

같은 문자열을 찾는 문제는 string 정렬을 하고 난 뒤 찾는 것이 쉽다.

따라서, 접두어를 찾는 해당 문제에서도 sort()를 사용한다.

 

해당 블록을

for (int j = 0; j < phone_book[i-1].length(); j++) {
            
    if (phone_book[i-1][j] != phone_book[i][j]) break; 
	else cnt++;
}

 

아래와 같이 substr을 사용해서 비교하면 for문을 사용하지 않아도 돼 코드가 더 간단해진다.

substr(0, phone_book[i].length());

 

10분 정도 소요해서 풀었지만 cnt 초기화를 시키지 않아 TC 11번에서 오류가 있었다.

sort한 후 첫번째가 아닌 문자열이 접두어가 되는 경우에 발생한다.

for 문 들어갈 때 cnt를 0으로 초기화하는 것을 잊지 말자!

 

해당 문제 링크

https://programmers.co.kr/learn/courses/30/lessons/42577#

코딩테스트 연습 - 전화번호 목록 | 프로그래머스