본문 바로가기
프로그래머스/코딩테스트 고득점 kit

프로그래머스 해시 LV2 - 전화번호 목록

by 맴썰 2021. 8. 20.

문제 설명

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

  • 구조대 : 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입니다.

 


첫번째 풀이 - 정확성 만점 + 효율성 0점 = 불합격

import java.util.*;
class Solution {
    public boolean solution(String[] phone_book) {
        boolean answer = true
        String temp = "";
        for(int i=0; i<phone_book.length; i++){
            for(int j=i+1;j<phone_book.length;j++){
               if(phone_book[i].length()>phone_book[j].length()){
                   temp = phone_book[i];
                   phone_book[i] = phone_book[j];
                   phone_book[j] = temp;
               }
            }
        }for(int i=0; i<phone_book.length; i++){
            for(int j=i+1;j<phone_book.length;j++){
                if(phone_book[i].equals(phone_book[j].substring(0,phone_book[i].length())))
                    answer = false;
           }
        }
        return answer;
    }
}

시간복잡도 생각 안하고 푼 코드, 문자열 길이가 짧은 순서대로 정렬하고 모든 항목에 대해 해당 항목의 값을 접두어로 가지고 있는지 비교하는 코드. 

이제 시간복잡도를 줄이는 방법을 생각해봐야겠다.

 

두번째 풀이 - 첫번째와 동일

import java.util.*;
class Test implements Comparator<String>{

@Override
public int compare(String o1, String o2) {
if(o1.length() > o2.length()) return 1;
else if (o1.length() == o2.length()) return 0;
else return -1;
}
}

class Solution {
    public boolean solution(String[] phone_book) {
        boolean answer = true;
       
        String temp = "";
        Arrays.sort(phone_book, new Test());

     for(int i=0; i<phone_book.length; i++){
            for(int j=i+1;j<phone_book.length;j++){
                if(phone_book[j].startsWith(phone_book[i]))
                    answer = false;
               }
        }
        return answer;
    }
}

이번에는 문자열 길이 기준으로 정렬을 시킨 후 startsWith 함수를 사용하였지만 동일하게 효율성 문제가 발생하였다.

아무래도 2중 for문의 O(N^2)을 없애야 제대로 채점이 될 것 같다.

 

세번째 풀이 - 100점

import java.util.*;
class Test implements Comparator<String>{

@Override
public int compare(String o1, String o2) {
if(o1.length() > o2.length()) return 1;
else if (o1.length() == o2.length()) return 0;
else return -1;
}
}

class Solution {
    public boolean solution(String[] phone_book) {
        boolean answer = true;
        HashMap<String,Integer> map = new HashMap<String,Integer>();
        String temp = "";
        Arrays.sort(phone_book, new Test());
        
        for(int i =0;i<phone_book.length;i++) map.put(phone_book[i],phone_book[i].length());
        for(int i=0; i<phone_book.length; i++){
            for(int j=1; j<phone_book[i].length(); j++){
                if(map.containsKey(phone_book[i].substring(0,j))) 

                           return false;
               }
        }
        return answer;
    }
}

 

이번에는 HashMap을 사용해 정렬된 배열의 값과 문자열 길이값을 각각 Key,Value값으로 넣고, 

2중 for문의 반복 내용을 다르게 했다. 이전에는 문자열 배열을 각각 반복했다면, 지금은 첫번째 for문은 전체를 반복하고, 두번째 for문은 윗 for문이 가르키는 문자열의 길이만 반복하도록 하고, HashMap의 containsKey 메소드를 이용해 

1번째 for문의 0~j까지의 부분 문자열을 Key로 가지고 있는 값이 있는지 확인하고 있는 경우 false를 return 하도록 했다.

 

Value를 확인하고 싶으면 같은 방법으로 하되 containsValue 메소드를 사용하면 된다. 

 

HashMap과 HashSet을 더 공부해야겠다.