회문 성공

1 초 (추가 시간 없음) 512 MB 15091 4131 2961 28.246%
문제

회문(回文) 또는 팰린드롬(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를 순서대로 한 줄에 하나씩 출력한다.
예제 입력 1 복사
7
abba
summuus
xabba
xabbay
comcom
comwwmoc
comwwtmoc
예제 출력 1 복사
0
1
1
2
2
0
1

예외케이스를 똑바로 처리하지 않아 10번이나 틀린 화가 잔뜩나는 문제이다.

문제는 간단하다, 문자열의 시작점을 첫번째 포인터, 끝점을 두번째 포인터로 잡아 첫번째 포인터 <= 두번째 포인터 가 참일 동안 계속 확인해주면 된다.

다만 여기서 유사회문이라는 것을 판별해주어야 하는데, 이것은 문자열에서 한가지 문자를 제거하면 회문이 되는 문자열이다.

이는 첫번째 포인터 + 1 == 두번째 포인터, 첫번째 포인터 == 두번째 포인터 - 1 인 경우에만 성립할 수 있다.

만일 이것이 성립하지 않으면 유사회문도, 회문도 아닌것이다.

나는 이 문제를 풀때 순차적으로 조건문을 써 return시켜주어 첫번째 조건에 해당하지 않는데 두번째 조건에 해당하는 경우를 잡아내지 못하였다.

시간복잡도도 계산하기 좀 힘들었다.

#include <iostream>
#include <string>
using namespace std;
string s;
int T;
int func(int st, int last, int ans){
        while(st <= last){
            if(s[st] == s[last]){
                st += 1;
                last -= 1;
                continue;
            }
            if(ans == 1){
                return 2;
            }
            int case1 = func(st+1, last, 1);
            int case2 = func(st, last-1, 1);
            return min(case1, case2);
        }
        return ans;
}
int main(){
    cin>>T;
    while(T--){
        int ans = 0;
        cin>>s;
        int st = 0, last=s.size()-1;
        cout<<func(st, last, 0)<<'\n';
    }
}

코드를 보면 int형 func를 하나 만들어 return값이 곧바로 정답이 되게 하였다.

string이 그냥 회문일 경우는 1/2N(size of string) 이 걸리게 되는데,

여기서 재귀가 돌아가면서 일단 회문이 아니면 N타임이 걸리게 된다 ( st+1한 func, last -1한 func (1/2N * 2))

하지만 ans == 1일때를 조건으로 걸어, 이 문제는 O(N)으로 풀 수 있게 만들었다.