[백준] 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를 순서대로 한 줄에 하나씩 출력한다.

내 풀이

def Is_palindrome(str1):
    p = str1[::-1]
    if p == str1:
        print(0)    # palindrome
    else:
        subwords = []
        for i in range(len(str1)):
            a = str1[0:i]
            b = str1[i+1:]
            c = a+b
            subwords.append(c)
            
        sub_pal = [(x[::-1], x) for x in subwords]
        real = [x for x in sub_pal if x[0]==x[1]]

        if len(real) >= 1:    # pseudo-palindrome
            print(1)
        else:      # not palindrome
            print(2)
    
input1 = int(input())
for a in range(input1):
    Is_palindrome(input())

메모리 초과로 실패....
유사 팰린드롬을 판단하는 과정에서 문자열을 많이 만들어버려서 그런 듯...

투 포인터 문제인 것을 알고 다시 아래와 같이 접근해보았다. (250321) 그러나 아래 코드도 어디에 한 문자가 있는지를 고려하지 못해서 오답을 받게 된다.

string = "xabba"
rest = []
for i in range(len(string)//2):
    if string[i] == string[-(i+1)]:
        pass
    else:
        rest.append(string[i])
if len(rest)==1:
    print(1)
else:
    print(2)

진짜 정답은 아래 코드이다. (코멘트 참조)

def is_palindrome(string, left, right):
    while left < right:
        if string[left] != string[right]:
            return False
        left += 1
        right -= 1
    return True

def pseudo_palindrome(string):
    left, right = 0, len(string)-1
    while left < right:
        if string[left] != string[right]:
            if is_palindrome(string, left+1, right) or is_palindrome(string, left, right-1):
                return 1
            else:
                return 2     
        left += 1
        right -= 1    
    
            
t = int(input())
for _ in range(t):
    string = input()
    if is_palindrome(string, 0, len(string)-1):
        print(0)
        continue
    else:
        print(pseudo_palindrome(string))      

코멘트

보다 효율적으로 (유사) 회문을 판단하려면 문장 처음과 끝에 포인터를 설정하고 문장의 가운데로 좁혀가면서 체크하는 방법을 사용하면 된다.
게다가 정답인 풀이는 각각 왼쪽 또는 오른쪽으로부터 한 문자를 삭제해가면서 회문인지 체크하기 때문에 방해가 되는 문자가 문자열의 어느 부분에 위치하든지 잡아낼 수 있다.

References

https://www.acmicpc.net/problem/17609
https://aldrn29.tistory.com/32