[Programmers] 전화번호 목록 (Java)

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

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

문제

입력값으로 문자열 배열이 주어집니다. 각 문자열들은 전화번호부에 적힌 전화번호를 나타냅니다. 여기서 각 전화번호가 서로 다른 전화번호의 접두어가 되는 경우가 있는지 확인하여 반환하면 되는 문제입니다. 어떠한 번호가 다른 번호의 접두어가 되면 false를, 그렇지 않으면 true를 반환합니다.

 

접근 아이디어

1. 자료구조 Set을 이용하여 각 전화번호의 접두어가 될 수 있는 경우를 모두 저장합니다.

2. 실제 각 전화번호들이 Set에 저장된 접두어들에 포함이 되는지를 확인하면 됩니다.

 

조심해야할 점

아래 코드에서 Set에 저장시키는 접두어를 만들 때 j의 범위에 주의해야 합니다. 위의 2번째 과정을 진행할 때, 접두어를 체크해야 하는데 만약 각 전화 번호의 전체 범위까지도 Set에 저장되어 있으면 무조건 false를 반환하게 됩니다.

 

- Java

import java.util.*;

class Solution {
   public boolean solution(String[] phone_book) {
        
        boolean answer = true;
        
        HashSet<String> set = new HashSet<>();
        for(int i=0; i<phone_book.length; i++) {
        	String phone = phone_book[i];
        	for(int j=1; j<phone.length(); j++) {
        		set.add(phone.substring(0,j));
        	}
        }
        
        for(int i=0; i<phone_book.length; i++) {
        	if(set.contains(phone_book[i])) {
        		answer = false;
        		break;
        	}
        }
        
        return answer;
    }
}

 

 

Java 채점 결과