BOJ-5052 전화번호 목록 Swift

문제

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

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

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

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

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

입력

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

출력

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

내가 푼 풀이

- 전화번호가 차례로주어지는것이 아니라 이미 목록에 있고, 정렬할 수 있다.

- 전화번호를 숫자형식으로 오름차순 정렬하는것과 문자형식으로 정렬하는것이 결과가 다르다.

"""
113
12340
123440
12345
98346
문자열배열 오름차순정렬: ["113", "12340", "123440", "12345", "98346"]
숫자배열 오름차순정렬:   [113, 12340, 12345, 98346, 123440]
"""

- 이전의 번호가 접두사로 존재한다면 이 목록은 일관성이 없다고 한다.

- 숫자배열과 다르게 문자열배열의 정렬은 각각 자리마다의 문자를 비교하므로 접두사를 찾기에 더 알맞은 정렬방법이 된다.

- 이렇게 정렬된 문자열배열을 순회하며 i+1번째 문자열의 접두사가 i번째문자인지만 파악하면 된다.

 

import Foundation

// 입력받기
let t = Int(readLine()!)!

for i in 0..<t {
    let n = Int(readLine()!)!
    var str = [String]()
    var result = true
    for j in 0..<n {
        let num = readLine()!
        str.append(num)
    }
    str.sort(by: <)
    for j in 0..<str.count-1 {
        if str[j+1].hasPrefix(str[j]) {
            result = false
            break
        }
    }
    if result {
        print("YES")
    } else {
        print("NO")
    }
}