[Algospot] 고대어 사전 (문제 ID : DICTIONARY)

문제

아마추어 고고학자인 일리노이 존스는 시카고 근교에서 고대 문명의 흔적을 찾아냈습니다. 그 흔적 중에는 이 언어의 사전도 포함되어 있었는데, 이 사전에 포함된 단어들은 모두 영어의 소문자 알파벳으로 구성되어 있었지만 사전에 포함된 단어의 순서들이 영어와 서로 달랐습니다. 발굴팀은 단어들이 사전 순이 아닌 다른 순서대로 정렬되어 있는지, 아니면 알파벳들의 순서가 영어와 서로 다른 것인지를 알고 싶어합니다.

일리노이 존스는 이 언어에서는 알파벳들의 순서가 영어와 서로 다를 뿐, 사전의 단어들은 사전 순서대로 배치되어 있다는 가설을 세웠습니다. 이 가설이 사실이라고 가정하고, 단어의 목록으로부터 알파벳의 순서를 찾아내려고 합니다.

예를 들어 다섯 개의 단어 gg, kia, lotte, lg, hanhwa 가 사전에 순서대로 적혀 있다고 합시다. gg가 kia보다 앞에 오려면 이 언어에서는 g가 k보다 앞에 와야 합니다. 같은 원리로 k는 l앞에, l은 h앞에 와야 한다는 것을 알 수 있지요. lotte 가 lg 보다 앞에 오려면 o가 g 보다 앞에 와야 한다는 것도 알 수 있습니다. 이들을 종합하면 다섯 개의 알파벳 o, g, k, l, h 의 상대적 순서를 알게 됩니다.

사전에 포함된 단어들의 목록이 순서대로 주어질 때 이 언어에서 알파벳의 순서를 계산하는 프로그램을 작성하세요.

 

입력

입력의 첫 줄에는 테스트 케이스의 수 C (1 <= C <= 50) 가 주어집니다. 각 테스트 케이스의 첫 줄에는 사전에 포함된 단어의 수 N (1 <= N <= 1000) 이 주어집니다. 그 후 N 줄에 하나씩 사전에 포함된 단어가 순서대로 주어집니다. 각 단어는 알파벳 소문자로 구성되어 있으며, 길이는 1 이상 20 이하입니다. 중복으로 출현하는 단어는 없습니다.

 

출력

각 테스트 케이스마다 한 줄을 출력합니다. 만약 알파벳들의 순서에 모순이 있다면 "INVALID HYPOTHESIS"를 출력하고, 모순이 없다면 26개의 소문자로 알파벳들의 순서를 출력합니다. 만약 가능한 순서가 여러 개 있다면 아무 것이나 출력해도 좋습니다.

---

dfs 문제이다.

우선 들어온 문자열들을 이용해 그래프를 만든다. 그래프의 정보는 몇 번째 알파벳이 몇 번째 알파벳보다 먼저 와야하는가를 저장한다.

그래프를 모두 만들었으면, 그 그래프를 토대로 알파벳 순서를 저장한다. 방문하지 않은 점 중에 x 다음으로 들어올 글자라면 방문한다.

dfs인지라 글자가 거꾸로 저장되기 때문에, 나중에 reverse를 해줘야 함을 유의하자.

그 후엔 그 결과 문자열에 모순이 있는지를 확인한 후 모순이 있다면 invalid 를 출력하고 아니라면 결과를 출력한다. 

 

 

코드 원본 : https://github.com/chosh95/STUDY/blob/master/Algospot/DICTIONARY.cpp

chosh95/STUDY

C++ 코드

#include <iostream>
#include <cstring>
#include <vector>
#include <algorithm>
using namespace std;
int C, N;
int p[27][27]; //p[i][j] : 알파벳 i가 j보다 먼저와야 한다.
bool check[27];
vector<string> str;
vector<char> res;

void makeGraph() {
	for (int j = 1; j < N; j++) {
		int i = j - 1;
		int len = min(str[i].length(), str[j].length());
		for (int k = 0; k < len; k++) {
			if (str[i][k] == str[j][k]) continue;
			int a = str[i][k] - 'a';
			int b = str[j][k] - 'a';
			p[a][b] = 1;
			break;
		}
	}
}

void dfs(int x) {
	check[x] = true;
	for (int nx = 0; nx < 26; nx++) {
		if (p[x][nx] == 1 && !check[nx])
			dfs(nx);
	}
	res.push_back(char(x + 'a'));
}

bool makeResult() {
	reverse(res.begin(), res.end());
	for (int i = 0; i < 26; i++) {
		for (int j = i+1; j < 26; j++) {
			if (p[(int)(res[j] - 'a')][(int)(res[i] - 'a')] == 1)
				return false;
		}
	}
	return true;
}
int main()
{
	cin >> C;
	while (C--) {
		cin >> N;

		str.resize(N);
		res.clear();
		memset(p, 0, sizeof(p));
		memset(check, false, sizeof(check));

		for (int i = 0; i < N; i++) cin >> str[i];
		makeGraph();
		for (int i = 0; i < 26; i++) {
			if (!check[i]) dfs(i);
		}

		if (!makeResult())
			cout << "INVALID HYPOTHESIS\n";
		else {
			for (int i = 0; i < 26; i++)
				cout << res[i];
			cout << "\n";
		}

	}
}