취미로 코딩하기

※ 글쓴이는 취미로 코딩을 익혀보는 사람이라 정확하지 않은 내용을 담고 있을 수 있다 ※

이번에 볼 문제는 백준 14010번 문제인 Where To Go?이다.
문제는 아래 링크를 확인하자.

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

길이가 같은 대문자 문자열 A와 소문자 문자열 B의 각 문자를 일대일 대응시킬 수 있는지를 확인할 수 있다면, B와 같은 길이의 A의 부분문자열과 B에 대해 각각 대응이 가능한지를 시도해보는 것으로 문제를 해결할 수 있을 것이다.

A를 읽어나가면서 A에서 처음 등장하는 알파벳이라면 B의 대응되는 순서의 알파벳과 대응시키고, 이미 등장한 알파벳이라면 기존 대응된 B의 알파벳과 B의 대응되는 순서의 알파벳이 같은지 체크하는 것으로 A에서 B로의 대응이 가능한지를 확인할 수 있다.

B에서 A로의 대응이 가능한지도 점검하면 A와 B의 각 문자가 일대일대응이 가능한지를 확인할 수 있다.

이를 이용해 문제를 해결하자.

아래는 제출한 소스코드이다.

#include <iostream>
#include <string>
#include <cstring>
using namespace std;

int arr[128];

int main() {
	ios::sync_with_stdio(0);
	cin.tie(0);
	
	int T; cin >> T;
	string s; cin >> s;
	int slen = s.length();

	while (T--) {
		bool chk1 = 1;
		string ss; cin >> ss;
		int sslen = ss.length();
		for (int i = 0; i <= slen - sslen; i++) {
			bool chk2 = 1;
			memset(arr, 0, sizeof(arr));
			for (int j = 0; j < sslen; j++) {
				if (arr[ss[j]]) {
					if (arr[ss[j]] != s[i + j]) {
						chk2 = 0;
					}
				}
				else arr[ss[j]] = s[i + j];
			}
			for (int j = 0; j < sslen; j++) {
				if (arr[s[i + j]]) {
					if (arr[s[i + j]] != ss[j]) {
						chk2 = 0;
					}
				}
				else arr[s[i + j]] = ss[j];
			}

			if (chk2) {
				cout << i << '\n';
				chk1 = 0;
				break;
			}
		}

		if (chk1) cout << '-' << '\n';
	}
}