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

문제 설명

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

• 구조대 : 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

Solution

C++

#include <string>
#include <vector>
#include <unordered_map>

using namespace std;

bool solution(vector<string> phone_book) {
    bool answer = true;
    string str = "";
    unordered_map<string, int> d;
    
    for(auto i : phone_book){
        str = "";
        for(int j=0; j<i.size(); j++){
            str += i[j];
            d[str]++;
        }
    }
    
    for(auto i : phone_book){
        if(d[i] > 1){
            answer = false;
            break;
        }
    }
    return answer;
}

이 문제도 마찬가지로 unordered_map을 활용한다. hash 알고리즘을 우리가 직접 개발 할 필요는 없으나, STL 중에서 hash를 써먹는 것들을 이용해서 좀 더 빠른 데이터 처리를 기대할 수는 있다.

우선 phone_book vector에 있는 데이터들을 for_each를 통해 끌고온다. 그리고 각각의 데이터들을 가져오되, 각각 데이터들의 부분들(?)을 모두 unordered_map에 저장한다.
무슨 의미냐면 119 라는 데이터를 집어넣는다면 1, 11, 119 이렇게 세 가지 데이터를 각각 unordered_map에 집어넣는다는 의미이다.

그리고 다시 phone_book vector를 뒤져보며 각 데이터들 중에 하나라도 겹치게 들어간 녀석이 있는지를 확인한다.
119라는 데이터를 1, 11, 119 이렇게 저장했다면 1195524421 라는 데이터 또한 그렇게 저장했을 것이고 119 가 겹치게 들어갔다. 즉 119를 접두사로 하는 데이터가 존재한다는 의미다.

Hashing 기법을 연습하는 데 상당히 도움이 되는 문제였다. unoredered_map을 활용하는 방법을 조금 더 연구해볼 수 있었던 문제였다.

Java

import java.util.HashMap;

class Solution {
     public boolean solution(String[] phone_book){
        boolean answer = true;
        HashMap<String, Integer> map = new HashMap<>();
        String str = "";
        for(String i : phone_book){
            str = "";
            for(int j = 0; j < i.length(); j++){
                str += i.charAt(j);
                map.put(str, map.getOrDefault(str, 0) + 1);
            }
        }
        for(String i : phone_book){
            if(map.get(i) > 1){
                answer = false;
                break;
            }
        }

        return answer;
    }
}

마찬가지로 HashMap을 사용한다. 원리는 똑같다. 그저 문법의 차이일 뿐.

Python

def solution(phone_book):
    phone_book = sorted(phone_book)
    for p1, p2 in zip(phone_book, phone_book[1:]):
        if p2.startswith(p1):
            return False
    return True

파이썬의 강력함을 체감할 수 있는 순간이다. 그냥 단순하게 zip 함수로 묶어버리고 for문을 돌려버렸다.

zip 함수에 대해 조금 설명하자면 두 가지 데이터를 마치 지퍼 잠그듯 묶어버린다는 의미를 가지고 있다.

>>> numbers = [1, 2, 3]
>>> letters = ["A", "B", "C"]
>>> for pair in zip(numbers, letters):
>>>    print(pair)

(1, 'A')
(2, 'B')
(3, 'C')

주 언어가 파이썬이 아니다보니 좀 어색하긴 한데, 이 함수를 활용해서 적절한 데이터를 생성했다.

한번 시험삼아 출력을 해 보았다. 어쨌거나 중요한 건, 특정한 값이 어떤 값의 접두사냐 아니냐다는 것이다. 모든 경우에 대해 대입 해 보면 답이 금방 나올 수도 있다.