[BOJ] 17609 회문 (python)

문제

회문(回文) 또는 팰린드롬(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를 순서대로 한 줄에 하나씩 출력한다.

풀이

계속 시간초과에 걸리다가 겨우 풀었다.
원래는 while문 안에 if palin == palin[::-1] 조건문을 넣어뒀는데, 백준 질문 목록을 보니 slicing은 시간 복잡도를 꽤 많이 잡아먹는다고 해서 밖으로 꺼내서 풀게 됐다.

그리고 처음에는 recheck(palin, start, end)없이 while 안에서 인덱스만 증가시키고 계속 진행하는 방식을 택했는데 aapqbcbqpqaa라는 반례에 막혔고, 결국에는 따로 함수로 만들어서 풀게 됐다. 사실 while안에 넣어둬도 되는데 너무 보기 싫어져서 따로 뺐다.

import sys

T = int(sys.stdin.readline())

def recheck(palin, start, end):
    while start < end:
        if palin[start] == palin[end]:
            start += 1
            end -= 1
        else:
            return False
    return True

for i in range(T):
    palin = sys.stdin.readline().rstrip()
    start = 0
    end = len(palin)-1
    is_palindrome = 0
    
    if palin == palin[::-1]:
        print(0)
    else:
        while start < end:
            if is_palindrome >= 2:
                break
            if palin[start] == palin[end]:
                start += 1
                end -= 1
            else:
                if recheck(palin, start+1, end) or recheck(palin, start, end-1):
                    is_palindrome = 1
                    break
                else:
                    is_palindrome = 2
                    break
        print(is_palindrome)