[백준] 12871번 - 무한 문자열 (C++)

12871번: 무한 문자열

문제

문자열 s가 있을 때, f(s)는 s를 무한번 붙인 문자열로 정의한다. 예를 들어, s = "abc" 인 경우에 f(s) = "abcabcabcabc..."가 된다.

다른 문자열 s와 t가 있을 때, f(s)와 f(t)가 같은 문자열인 경우가 있다. 예를 들어서, s = "abc", t = "abcabc"인 경우에 f(s)와 f(t)는 같은 문자열을 만든다.

s와 t가 주어졌을 때, f(s)와 f(t)가 같은 문자열을 만드는지 아닌지 구하는 프로그램을 작성하시오.

 

 

입력

첫째 줄에 s, 둘째 줄에 t가 주어진다. 두 문자열 s와 t의 길이는 50보다 작거나 같은 자연수이고, 알파벳 소문자로만 이루어져 있다. 

 

출력

첫째 줄에 f(s)와 f(t)가 같으면 1을, 다르면 0을 출력한다.

 

 

풀이과정

문제에서 요구하는 사항들과 반례들은 아래와 같다.

s와 t를 무한하게 늘어트려두었을때 두개의 문자열이 같은 경우 1, 다른경우 0

1. f(s) = ab, f(t) = abab / output : 1

2. f(s) = aa, f(t) = aaa / output : 1

3. f(s) = abc, f(t) = abca / output : f(s) = 0 -> abcabc... , f(t) = abcaabca...

1번과 2번은 어렵지않게 접근할 수 있었는데 3번에 경우 2번과 반례 처리가 겹쳐 어떻게할까 고민을 하다가 그냥 문자열을

s * 2 와 t * 2를 서로 비교하는 방식으로 문제을 해결하였다.

 

code

#include <iostream>
#include <string>

void	strCheck(std::string str1, std::string str2)
{
	int len = 0;

	for (int i = 0; i < str1.length(); i++) {
		if (len == str2.length()) len = 0;
		if (str1[i] == str2[len]) len++;
		else {
			std::cout << "0" << "\n";
			return ;
		}
	}
	std::cout << "1" << "\n";
	return ;
}

int		main(void)
{
	std::string s, t;

	std::cin >> s >> t;
	if (s.length() <= t.length()) strCheck(t + t, s + s);
	else strCheck(s + s, t + t);

	return (0);
}

 

후기

문제자체는 간단했지만 나름 재미있었다.