[5052] 전화번호 목록

문제 링크 : https://www.acmicpc.net/problem/5052

5052 - 전화번호 목록

본문

전화번호 목록이 주어진다. 이때, 이 목록이 일관성이 있는지 없는지를 구하는 프로그램을 작성하시오.

전화번호 목록이 일관성을 유지하려면, 한 번호가 다른 번호의 접두어인 경우가 없어야 한다.

예를 들어, 전화번호 목록이 아래와 같은 경우를 생각해보자

• 긴급전화: 911
• 상근: 97 625 999
• 선영: 91 12 54 26

이 경우에 선영이에게 전화를 걸 수 있는 방법이 없다. 전화기를 들고 선영이 번호의 처음 세 자리를 누르는 순간 바로 긴급전화가 걸리기 때문이다. 따라서, 이 목록은 일관성이 없는 목록이다.

입력

첫째 줄에 테스트 케이스의 개수 $t$가 주어진다. ($1 \leq t \leq 50$) 각 테스트 케이스의 첫째 줄에는 전화번호의 수 $n$이 주어진다. ($1 \leq n \leq 10000$) 다음 $n$개의 줄에는 목록에 포함되어 있는 전화번호가 하나씩 주어진다. 전화번호의 길이는 길어야 10자리이며, 목록에 있는 두 전화번호가 같은 경우는 없다.

출력

각 테스트 케이스에 대해서, 일관성 있는 목록인 경우에는 YES, 아닌 경우에는 NO를 출력한다.

제한

시간 제한	메모리 제한
1sec	256MB

---

풀이

대놓고 트라이 자료구조를 활용할 것을 요구하는 문제다. 전화번호의 길이가 크지 않으므로, 한 문자마다 하나의 노드를 사용하도록 트라이 구조를 만들면 쉽게 해결할 수 있는 문제다. 각 노드는 0 ~ 9까지, 그리고 그 곳에서 전화번호가 끝난다는 None까지 총 11칸을 가지도록 하면 된다.

트라이 구조를 사용하지 않고 해결하는 방법으로는 “문자열 정렬”을 통해 해결할 수 있다. 문자열의 개수가 적고 길이 또한 길지 않아서 정렬에 큰 어려움은 없다. 문자열을 사전순으로 정렬한 이후, 모든 전화번호에 대해 전화를 걸 수 있는지 확인하는 것으로 문제를 해결할 수 있다. 전화번호를 걸 수 있는지의 여부는, 정렬되어있는 전화번호들에 대해 본인과 그 이전 번호 2개씩만 확인하면 된다. 이전 번호가 현재 번호의 prefix, 즉 접두사라면 전화를 걸 수 없다. 그 이외는 모두 전화를 걸 수 있는 상태.

즉, 문자열을 사전순으로 정렬하는 단계, 그리고 앞의 번호가 뒤의 번호의 prefix인지 확인하는 단계 해서 총 2단계를 구현하면 답을 얻을 수 있다.

• 참고 알고리즘 :

코드

사용 언어 : C

최종 수정일 : 2024 / 4 / 12

#include <stdio.h>

typedef char bool;
const bool true = 1;
const bool false = 0;

#define MAX_IDX 10000
#define MAX_LEN 10

int n;
char arr[MAX_IDX][MAX_LEN + 1];
char printTemplate[2][4] = {"NO", "YES"};

int cmp(char* a, char* b) {
    for (int i = 0; a[i] != '\0'; ++i) {
        if (a[i] != b[i]) {
            return a[i] > b[i];
        }
    }

    return 0;
}

int main() {
    int t;
    scanf("%d", &t);
    while (t--) {
        scanf("%d", &n);
        for (int i = 0; i < n; ++i) {
            scanf("%s", arr[i]);
        }
        qsort(arr, n, sizeof(char[MAX_LEN + 1]), cmp);

        bool isConsistent = true;
        for (int i = 1; i < n; ++i) {
            bool isNotSame = false;
            for (int j = 0; arr[i - 1][j] != '\0'; ++j) {
                if (arr[i - 1][j] != arr[i][j]) {
                    isNotSame = true;
                    break;
                }
            }

            if (isNotSame == false) {
                isConsistent = false;
            }

            if (isConsistent == false) {
                break;
            }
        }

        printf("%s\n", printTemplate[isConsistent]);
    }
    return 0;
}