[백준] 1259: 팰린드롬수 (Python)

팰린드롬수

• https://www.acmicpc.net/problem/1259

1259번: 팰린드롬수

 

문제

어떤 단어를 뒤에서부터 읽어도 똑같다면 그 단어를 팰린드롬이라고 한다. 'radar', 'sees'는 팰린드롬이다.

수도 팰린드롬으로 취급할 수 있다. 수의 숫자들을 뒤에서부터 읽어도 같다면 그 수는 팰린드롬수다. 121, 12421 등은 팰린드롬수다. 123, 1231은 뒤에서부터 읽으면 다르므로 팰린드롬수가 아니다. 또한 10도 팰린드롬수가 아닌데, 앞에 무의미한 0이 올 수 있다면 010이 되어 팰린드롬수로 취급할 수도 있지만, 특별히 이번 문제에서는 무의미한 0이 앞에 올 수 없다고 하자.

 

입력

입력은 여러 개의 테스트 케이스로 이루어져 있으며, 각 줄마다 1 이상 99999 이하의 정수가 주어진다. 입력의 마지막 줄에는 0이 주어지며, 이 줄은 문제에 포함되지 않는다.

 

출력

각 줄마다 주어진 수가 팰린드롬수면 'yes', 아니면 'no'를 출력한다.

 

입출력 예제

입력	출력
121 1231 12421	yes no yes

 

풀이

숫자를 문자열로 변환하여 입력 받은 후 투 포인터를 통해 양 끝의 원소가 같은지 비교하여 해결한다.

 

접근할 인덱스 구하기

• 길이가 5면 0-1, 3-4 / 길이가 6이면 0-2/3-5 / 길이가 7이면 0-2/4-7
• 두 포인터가 검사할 인덱스는 길이가 홀수면 0~(n//2 - 1) / (n//2 + 1)~(n-1), 길이가 짝수면 0~(n//2 - 1) / (n//2)~(n-1)
• → for i in range(0, n//2) 에 대해 num[i]와 num[n-1-i]가 같은지 비교 (ex. 0-4, 1-3, 2-2 / 0-5, 1-4, 2-3)

import sys
input = sys.stdin.readline
while True:
	num = input().rstrip()
	if num == '0':
		break
	flag = True
	n = len(num)
	for i in range(0, n//2):
		if num[i] != num[n-1-i]:
			flag = False
			break
	print("yes") if flag else print("no")

메모리: 31256 KB / 시간: 40 ms

시간 복잡도: O(n) - n // 2번 비교

공간 복잡도: O(1) - flag, n 변수 사용

 

투 포인터 대신 문자열을 뒤집어 해결하는 방법도 있다. 팰린드롬수는 앞에서부터 읽어도 뒤에서부터 읽어도 동일한 수이므로, 입력 받은 문자열과 그 문자열을 뒤집은 문자열이 동일하면 팰린드롬수라고 판단할 수 있다.

파이썬의 문자열에는 reverse() 메소드가 없으므로 리스트로 변환하여 뒤집고 다시 문자열로 바꿔야 하는 번거로움이 있다. 따라서 슬라이싱을 이용해 간단하게 뒤집는 방법을 선택했다.

• [python] 파이썬 문자열 뒤집기, 거꾸로 출력 하기 3가지 방법

import sys
input = sys.stdin.readline
while True:
	num = input().rstrip()
	if num == '0':
		break
	reversed_num = num[::-1]
	print("yes") if reversed_num == num else print("no")

메모리: 31256 KB / 시간: 40 ms

첫번째 코드와 성능은 동일하지만 코드 자체는 더 간결하다.