[17609] 회문

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

17609번: 회문

문제

회문(回文) 또는 팰린드롬(palindrome)은 앞 뒤 방향으로 볼 때 같은 순서의 문자로 구성된 문자열을 말한다. 예를 들어 ‘abba’ ‘kayak’, ‘reviver’, ‘madam’은 모두 회문이다. 만일 그 자체는 회문이 아니지만 한 문자를 삭제하여 회문으로 만들 수 있는 문자열이라면 우리는 이런 문자열을 “유사회문”(pseudo palindrome)이라고 부른다. 예를 들어 ‘summuus’는 5번째나 혹은 6번째 문자 ‘u’를 제거하여 ‘summus’인 회문이 되므로 유사회문이다.

여러분은 제시된 문자열을 분석하여 그것이 그 자체로 회문인지, 또는 한 문자를 삭제하면 회문이 되는 “유사회문”인지, 아니면 회문이나 유사회문도 아닌 일반 문자열인지를 판단해야 한다. 만일 문자열 그 자체로 회문이면 0, 유사회문이면 1, 그 외는 2를 출력해야 한다. 

입력

입력의 첫 줄에는 주어지는 문자열의 개수를 나타내는 정수 T(1 ≤ T ≤ 30)가 주어진다. 다음 줄부터 T개의 줄에 걸쳐 한 줄에 하나의 문자열이 입력으로 주어진다. 주어지는 문자열의 길이는 3 이상 100,000 이하이고, 영문 알파벳 소문자로만 이루어져 있다.

출력

각 문자열이 회문인지, 유사 회문인지, 둘 모두 해당되지 않는지를 판단하여 회문이면 0, 유사 회문이면 1, 둘 모두 아니면 2를 순서대로 한 줄에 하나씩 출력한다.

코드

#include <algorithm>
#include <cstring>
#include <iostream>
#include <map>
#include <set>
#include <stack>
#include <queue>
#include <vector>
// 16234
using namespace std;

int T;
string str;

bool isPalindrome(string s){
	int len = s.length();

	for(int i=0; i<len/2; ++i){
		if(s[i] != s[len-(i+1)]){
			return false;
		}
	}

	return true;
}

bool isPseudo(string s){
	int len = s.length();
	string temp = s;
	int k = 0;

	for(int i=0; i<len/2; ++i){
		if(s[i] != s[len-(k+1)]){
			s.erase(s.begin()+i);
			if(isPalindrome(s)){
				return true;
			}
			temp.erase(temp.begin()+(len-(k+1)));
			if(isPalindrome(temp)){
				return true;
			}

			return false;
		}
		k++;
	}

	return false; // 오류
}

int main(void){
	cin >> T;
	for(int i=0; i<T; ++i){
		cin >> str;

		if(isPalindrome(str)){
			cout << 0; // 회문
		}
		else if(isPseudo(str)){
			cout << 1;
		}
		else{
			cout << 2;
		}

		cout << '\n';
	}
}

좀 변형된 펠린드롬 문제이다.

유사회문을 어떻게 푸느냐에 달린 문제인데 일반적으로 풀면 시간초과가 난다. 

때문에 좀더 시간을 줄이고자 생각한 방법이 어짜피 딱 한개 문자만 없앨거면은 펠린드롬 확인하는 함수와 똑같이 하되, 만약 양옆문자가 다를 때 왼쪽 문자를 무시했을 때와 오른쪽 문자를 무시했을 때 중 하나라도 통과하고 나머지는 다 문자가 같으면 유사회문으로 지정할 수 있다.

 

s.erase(s.begin()+1)의 뜻은 s문자열의 첫번째 문자를 삭제시킨다는 것이다.