BAEKJOON #17609 회문 (문자열) - python

nathan·2021년 8월 26일
0

알고리즘문제

목록 보기
52/102

회문

출처 : 백준 #17609

시간 제한메모리 제한
1초512MB

문제

회문(回文) 또는 팰린드롬(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


풀이

생각 및 풀이 설명

  • 처음에 두 포인터를 통해 접근하고자 하였다.
  • 만약 원래의 string 리스트의 역순 리스트가 같지 않다면(회문이 아니라면), start=0, end=n-1로 두고, 양 쪽에서 한 칸씩 줄여가며 틀린 부분을 탐색했다.
    • 그리고 해당 부분을 찾는다면,
    • 1) 우선 start+=1일 때(맨 앞 원소 제외)를 고려하여 양쪽에서 똑같이 한 칸씩 줄여가며 틀린 부분을 탐색했다.
    • 2) 이 때 한 번 더 틀린 부분이 나온다면, "유사회문"의 가능성이 없으므로 end-=1을 통해 맨 뒤 원소를 제외하여 탐색했다.
    • 3) 만약 두 경우 모두 했을 때 틀린 부분이 한 번씩 더 나오면, "유사회문"의 가능성이 없으므로 2를 반환하였다.
    • 이 방법은 다음과 같은 반례를 통과하지 못한다.
      • abcddcdba
      • 이유는 1)에서 틀린 부분을 또 탐색하게 된다. (잘못된 원소를 제거해서..)
      • 사실 cddcd에서 첫 번째 원소가 아니라 마지막 원소를 제외해야하지만 순차적으로 작성한 코드는 1)로 들어갔을 때 틀린 부분을 탐색하게 되면 2를 반환하게 된다.
# 백준 17609번 회문 (틀린 코드)
def solution(string, count):
    n = len(string)
    r_string = list(reversed(string))
    if string == r_string:
        return 0
    else:
        if count < 1:
            start = 0
            end = n-1
            while start < end:
                if string[start] == string[end]:
                    start += 1
                    end -= 1
                else:
                    count += 1
                    if string[start+1] == string[end]:
                       start += 1
                    elif string[start] == string[end-1]:
                       end -= 1
                    else:
                        return 2
                    if count > 2:
                        return 2
        else:
            return 1
  • 아래의 코드는 위와 같은 문제점을 해결한 코드이다.
  • 우선 첫 string이 회문이 아닐 때, count를 점검한다.
  • count가 2보다 작다면(0 or 1), 포인터 start, end를 두어 탐색한다.
  • 만약 탐색하다 틀린 부분을 만났을 때(위의 방식과는 다른 점!!), 왼쪽 끝 원소를 제거한 경우와 오른쪽 끝 원소를 제거한 경우를 모두 구한다.(재귀를 통해)
  • 만약 재귀의 반환 값이 하나라도 0이라면 "유사회문"이므로 1을 반환하고, 그 외의 경우에는 "유사회문" 조차 아니므로 모두 2를 반환한다.

python code

# 백준 17609번 회문
from sys import stdin
input = stdin.readline

def solution(string, count):
    n = len(string)
    r_string = list(reversed(string))
    if string == r_string:
        return 0
    else:
        if count < 1:
            start = 0
            end = n-1
            while start < end:
                if string[start] == string[end]:
                    start += 1
                    end -= 1
                else:
                    count += 1
                    a = solution(string[start+1:end+1], count)  # 왼쪽 원소 하나 제외
                    b = solution(string[start:end], count)      # 오른쪽 원소 하나 제외
                    if min(a, b) == 0:
                        return 1
                    else:
                        return 2
        else:
            return 2    # 두 개 이상 원소 제외의 경우
                    


t = int(input())
answer = []
for i in range(t):
    string = list(map(str, input().rstrip()))
    result = solution(string, 0)
    answer.append(result)

for j in range(t):
    print(answer[j])
profile
나는 날마다 모든 면에서 점점 더 나아지고 있다.

0개의 댓글