본문 바로가기
  • Code Smell
Language/Algorithm

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

by HSooo 2021. 5. 14.

문제

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

구조대 : 119
박준영 : 97 674 223
지영석 : 11 9552 4421
전화번호부에 적힌 전화번호를 담은 배열 phone_book 이 solution 함수의 매개변수로 주어질 때, 어떤 번호가 다른 번호의 접두어인 경우가 있으면 false를 그렇지 않으면 true를 return 하도록 solution 함수를 작성해주세요.

제한 사항
phone_book의 길이는 1 이상 1,000,000 이하입니다.
각 전화번호의 길이는 1 이상 20 이하입니다.
같은 전화번호가 중복해서 들어있지 않습니다.
입출력 예제
["119", "97674223", "1195524421"] -> false
["123","456","789"] -> true
["12","123","1235","567","88"] -> false

배열의 한 인덱스의 값이 다른 인덱스가 포함하고 있는지를 찾는지 여부입니다.
이 문제는 효율성 트릭이 있어서 이중 for문으로 푸는 경우 실패합니다.

실패하는 코드 예시 (이중포문)

public boolean solution(String[] phone_book) {
    for (int i = 0; i < phone_book.length; i++) {
        String s = phone_book[i];
        for (int j = i+1; j < phone_book.length; j++) {
            if (i != j) {
                String k = phone_book[j];
                if (k.startsWith(s) || s.startsWith(k)) 
                    return false;
            }
        }
    }
    return true;
}

시간복잡도가 N^2이 되는건데, N^2 미만으로 나와야 통과하는 것 같습니다.

따라서 String 배열이기 때문에 오름차순으로 정렬할 경우 맨 앞 문자열부터 일치하는 순서대로 정렬 되므로 정렬 을 수행하고, index의 문자열이 index+1의 위치에 포함되는지 검사하면 될 겁니다.

수정한 코드

public boolean solution(String[] phone_book) {
    Arrays.sort(phone_book);
    for (int i = 0; i < phone_book.length-1; i++) {
        if (phone_book[i+1].startsWith(phone_book[i])) {
            return false;
        }
    }
    return true;
}

문자열 정렬의 경우 맨앞부터 일치하는 순서대로 정렬되므로 그걸 이용하면 될것 같습니다.

예) 문자열 [1, 5, 10, 13] 을 정렬하는 경우 -> [1, 10, 13, 5]
즉 1은 10과 13의 접두사가 됩니다. 그래서 n번째 값이 n+1번째 값의 접두사인지를 검사하면 됩니다.

'Language > Algorithm' 카테고리의 다른 글

[LeetCode] 2. Add Two Numbers  (0) 2021.08.26
[Codility] CyclicRotation (Java)  (0) 2021.03.24
[Programmers] 체육복 (Java)  (0) 2021.03.23
[Codility] BinaryGap (Java)  (0) 2021.03.23
[LeetCode] Climbing Stairs 문제 (memoization)  (0) 2021.02.10

댓글