[BOJ] 5052_전화번호 목록

전화번호 목록

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

5052번: 전화번호 목록

 

sort로 string을 정렬하면 사전순으로 정렬된다.
이를 이용하여 정렬 후 현재 데이터가 다음의 데이터 부분에 속하게 되면, 일관성이 유지되지 않았다는 것을 의미한다.

 

• 정답 코드

    #include<iostream>
    #include<algorithm>
    #include<vector>
    using namespace std;
    bool compare(string a, string b) {
        return a.size() < b.size();
    }
    int main() {
        ios_base::sync_with_stdio(false);
        cin.tie(NULL); cout.tie(NULL);
        int t;
        cin >> t;
        while (t--) {
            bool flag = true;
            int n;
            cin >> n;
            vector<string> list(n);
            for(int i = 0;i<n;i++) {
                cin >> list[i];
            }
            //사전순 정렬
            sort(list.begin(), list.end());
            for (int i = 0; (i < list.size() - 1) && flag; i++) {
                if(compare(list[i], list[i+1]))
                    if (equal(list[i].begin(), list[i].end(), list[i+1].begin())) {
                        flag = false;
                        break;
                    }
            }
            if (flag)
                cout << "YES\n";
            else
                cout << "NO\n";
        }
        return 0;
    }