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

 

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

알고리즘 문제 연습 카카오톡 친구해요! 프로그래머스 교육 카카오 채널을 만들었어요. 여기를 눌러, 친구 추가를 해주세요. 신규 교육 과정 소식은 물론 다양한 이벤트 소식을 가장 먼저 알려

school.programmers.co.kr


문제

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

구조대 : 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
입출력 예 설명
입출력 예 #1
앞에서 설명한 예와 같습니다.

입출력 예 #2
한 번호가 다른 번호의 접두사인 경우가 없으므로, 답은 true입니다.

입출력 예 #3
첫 번째 전화번호, “12”가 두 번째 전화번호 “123”의 접두사입니다. 따라서 답은 false입니다.


문제 해결을 위한 과정

이 문제는 번호 하나하나를 다른 번호와 일일이 비교하면 O(N^2)의 시간이 걸려 효율성 테스트를 통과할 수 없습니다. 따라서 Hash를 사용해서 문제를 풀어야 합니다.

  1. 전체 번호 해싱: 우선 모든 전화번호를 HashMap에 담습니다. 이때 getOrDefault를 사용하여 각 번호가 존재함을 기록합니다.
  2. 접두어 검사: 다시 전체 번호를 하나씩 꺼내어(phone), 해당 번호의 첫 글자부터 마지막 직전 글자까지 잘라보며(substring) 그 자른 문자열이 우리 Map에 존재하는지 확인합니다.
  3. 즉시 반환: 만약 자른 문자열이 Map에 있다면, 어떤 번호가 다른 번호의 접두어라는 뜻이므로 즉시 false를 반환하고 종료합니다.

키포인트

  1. 주의점: substring을 이용하여 map에 있는지 검사 시 범위를 phone.length() 직전까지 설정해야 자기 자신과 똑같은 번호를 찾는 실수를 방지할 수 있습니다.

소스코드
import java.util.*;

class Solution {
    static HashMap<String, Integer> map = new HashMap<>();
    
    public boolean solution(String[] phone_book) {
        boolean answer = true;
        
        for(String phone : phone_book) {
            map.put(phone, map.getOrDefault(phone, 0) + 1);
        }
        
        for(String phone : phone_book) {
            for(int i = 0; i < phone.length(); i++) {
                if(map.containsKey(phone.substring(0, i)))
                    return false;
            }
        }
        return answer;
    }
}

+ Recent posts