(백준 c++)5052 전화번호 목록

문제

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

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

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

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

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

입력

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

출력

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

 

 

#include <iostream>
#include <string>
#include <algorithm>
#include <vector>

using namespace std;

int main(void){
    int t;

    cin >> t;

    for(int i = 0; i < t; i++){
        int n;
        bool check = false;
        vector<string> phone;

        cin >> n;

        //전화번호 삽입
        for(int j = 0; j < n; j++){
            string s;
            cin >> s;
            phone.push_back(s);
        }

        //정렬
        sort(phone.begin(), phone.end());

        //정렬되었으므로 길이가 같은 경우는 서로 다른 전화번호이다.
        //하지만 앞에 있는 번호가 짧은 경우, 뒤에 있는 전화번호들을 비교해야 한다.
        //접두사이기 때문에 짧은 번호의 크기만큼 번호를 잘라서 비교를 한다.
        for(int j = 0; j < phone.size() - 1; j++){
            int current = phone[j].size();
            int compare = phone[j + 1].size();

            if(current == compare){
                continue;
            }
            else{
                if(phone[j + 1].substr(0, current) == phone[j]){
                    check = true;
                    break;
                }
            }
        }

        if(check){
            cout << "NO" << endl;
        }
        else{
            cout << "YES" << endl;
        }
    }
    
    return 0;
}