5052번: 전화번호 목록

문제 설명

https://www.acmicpc.net/problem/5052

5052번: 전화번호 목록

[문제]

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

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

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

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

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

 

[입력]

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

 

[출력]

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

 

[예제 입력 1]

2
3
911
97625999
91125426
5
113
12340
123440
12345
98346

[예제 출력 1]

NO
YES

---

풀이

 이 문제에 대한 알고리즘은 문제 설명에서 주어지는 행위를 일종의 알고리즘화(?)한 방법이다. 단순히 전화번호들을 특정 자료구조에 저장하고, 현재의 전화번호를 일정 길이로 잘라 해당 자료구조에서 찾을 때 만약 찾아지면 이는 일관성이 없는 것이고, 찾을 수 없다면 일관성이 있는 것으로 판단할 수 있다.

 이를 효율적으로 하기 위해서 우선적으로 입력 받은 전화번호들을 길이 순으로 짧은 전화번호가 먼저 오게 정렬한다. 그 다음 탐색을 시작하는데, 이때 가장 앞에 오는 전화번호의 길이를 가장 최소 사이즈로 삼고 해당 전화번호를 unordered_set에 저장한다. 이후 순차적으로 다음에 오는 전화번호를 최소 사이즈부터 하나씩 크기를 늘려 부분문자열로 잘라 이를 unordered_set에서 탐색을 한다. 만약 여기서 찾아진다면 일관성이 없으니 "NO"를 반환하고, 그렇지 않다면 해당 전화번호를 unordered_set에 저장하고 다음 전화번호에서 탐색을 시작한다. 이렇게 마지막 전화번호까지 끝났다면 최종적으로는 "YES"를 반환한다.

#include <iostream>
#include <algorithm>
#include <unordered_set>

using namespace std;

bool compare(string a, string b) {//단순 길이순 정렬
	return a.size() < b.size();
}

string search(int n, vector<string> phoneNumbers) {
	int minSize = phoneNumbers[0].size();

	unordered_set<string> phoneNumbersSet;

	phoneNumbersSet.insert(phoneNumbers[0]);//맨 처음 전화번호는 그냥 저장

	for (int i = 1; i < n; i++) {//이후의 전화번호들에 대해
		for (int j = minSize; j <= phoneNumbers[i].size(); j++) {//최소 사이즈부터 하나씩 늘리며
			string s = phoneNumbers[i].substr(0, j);

			if (phoneNumbersSet.find(s) != phoneNumbersSet.end())//앞선 전화번호들 중에 있다면
				return "NO";
		}

		phoneNumbersSet.insert(phoneNumbers[i]);//현 전화번호를 최종 저장
	}

	return "YES";
}

int main() {
	int t, n;
	string temp;
	vector<string> answer;

	cin >> t;

	for (int i = 0; i < t; i++) {
		vector<string> phoneNumbers;
		cin >> n;

		for (int j = 0; j < n; j++) {//전화번호 저장
			cin >> temp;
			phoneNumbers.push_back(temp);
		}

		sort(phoneNumbers.begin(), phoneNumbers.end(), compare);//정렬

		answer.push_back(search(n, phoneNumbers));
	}

	for (int i = 0; i < t; i++)
		cout << answer[i] << endl;

	return 0;
}

---

결과