[백준] 5052 전화번호 목록 + Trie 트라이

문제

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

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

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

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

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

입력

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

출력

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

 

 

자료구조 Trie(트라이)에 대해서 공부해봤습니다.

트리랑 비슷하게 루트에서 뻗어내려가는 형태입니다.

예를들어 "113"인 경우, 순서대로 root -> 1 -> 1 -> 3으로 뻗어내려가고, 3에서는 문자열이 끝나므로 표시를 합니다.

그런식으로 모든 문자열에 대해 반복하는데, 만약 접두배열이 같은경우, 같은 노드를 따라 내려옵니다.

 

이 문제에서는 일관성 있는 전화번호부를 물어봤습니다.

일관성이 있는 전화번호부란, 전화번호부의 어떤 전화번호가 다른 전화번호의 접두배열이면 안됩니다!

따라서 만약 어떤 마지막 노드가 아닌 어떤 노드에서 어떤 문자열이 종료가 되면 그 전화번호부는 일관성이 없습니다.

 

Trie의 구현은 다음과 같습니다.

 

	#include<bits/stdc++.h>
	#define FAST ios_base::sync_with_stdio(false),cin.tie(NULL);
	#define mset(v) memset(v,0,sizeof(v));
	#define rep(i,a) for(int i=0;i<a;++i)
	#define REP(i,a) for(int i=1;i<=a;++i)

	using namespace std;

	typedef long long ll;
	typedef pair<int, int> pi;
	typedef tuple<int, int, int>ti;
	typedef vector<int> vi;
	typedef vector<vector<int>> vvi;
	int dy[] = { -1,0,1,0 }, dx[] = { 0,1,0,-1 }, INF = 987654321;

	struct Trie {
		Trie* go[10]; //이 문제에서는 숫자 0~9까지 있으므로, 자식노드는 0~9까지 총 10개의 후보가 있습니다.
		bool output; //이곳에서 끝나는 전화번호가 있는지(더이상 문자가 없으면 true)
		bool goexit;  //자식 노드가 존재하는지

		//생성자
		Trie() {
			fill(go, go + 10, nullptr);
			output = goexit = false;
		}
		// 소멸자
		~Trie() {
			for (int i = 0; i < 10; i++)
				if (go[i]) delete go[i];
		}
		//insert 매소드
		void insert(string key) {//문자열 key를 입력받아서
			if (!key.length())output = true; //만약 key가 빈 문자열이면 종료합니다. output=true
			else {//아니라면
				int next = key[0]-'0';//문자열 맨 앞의 문자를 가리킵니다.
				if (!go[next])go[next] = new Trie; //만약 next에 대해 자식노드가 없을경우, 새로운 노드 생성
				goexit = true; //자식노드가 존재합니다. goexit = true
				go[next]->insert(key.substr(1)); //자식노드에 대해서 그 이후 문자열을 삽입합니다. 재귀의 형태입니다.
			}
		}

		bool consistent() { //일관성이 있는지 확인합니다.
			if (goexit && output) return false; //만약 자식노드가 존재하고, 이곳에서 끝나는 문자열이 존재하면 일관성이 있습니다.
			for (int i = 0; i < 10; i++) //자식노드들에 대해서 일관성이 있는지 확인합니다.
				if (go[i] && !go[i]->consistent()) return false;

			return true;//일관성이 있습니다.
		}
	};
	int main() {
		FAST;
		int T;
		cin >> T;
		while (T--) {
			int N;
			cin >> N;
			Trie* root = new Trie; // 루트 노드 만들기
			rep(i, N) {
				string tel;
				cin >> tel;
				root->insert(tel);//N개의 문자열을 입력하고 루트노드에 삽입합니다.
			} 
			root->consistent() ? cout << "YES\n" :cout<< "NO\n";
			delete root;
		}

	}