[백준] #1259 팰린드롬수(python)

📌문제

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

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

입력

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

출력

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

예제 입력

121
1231
12421
0

예제 출력

yes
no
yes

백준 1259번 문제

💡Idea

🤔 : 입력된 문자열을 어떻게 뒤집을 수 있을까?
👩‍💼 : 기러기 토마토 스위ㅅ....🙄

파이썬의 functools의 reduce라는 함수가 있습니다.
이 함수를 이용해서 입력된 문자열의 뒤에서부터 하나씩 더해가며 새로운 문자열을 만들어 비교하면 되겠다는 생각이 들었습니다.

💻코드

from functools import reduce
import sys
input = sys.stdin.readline

answer = []
while True:
    num = input().rstrip() # 입력된 숫자 문자열 뒤의 \n 제거 
    if num == '0': # 0을 입력받으면 종료
        break
    if num == reduce(lambda rev, cur: cur+rev, [n for n in num], ''): # 입력된 숫자와 뒤집은 숫자 비교
        answer.append('yes')
    else:
        answer.append('no')

print(*answer)

📝코드 설명

변수

• num : 입력되는 숫자 문자열을 하나씩 저장하는 변수
• answer : 각 수가 팰린드롬수인지 yes 혹은 no를 저장해두는 리스트

Reduce

reduce는 사실 어떤 수들의 합계를 찾을 때 많이 사용합니다.
지정된 초기값부터 시작하여 데이터를 순회하며 집계 함수를 적용함으로써, 데이터를 누적한 값을 구할 수 있습니다.

reduce(집계 함수, 순회 가능한 데이터[, 초기값])

집계 함수는 두 개의 인자를 받습니다.

1. 누적자(accumulator)
2. 현재 값(current value)

💻코드2

import sys
input = sys.stdin.readline

answer = []
while True:
    num = input().rstrip()  
    if num == '0': 
        break
    if num == num[::-1]:
        answer.append('yes')
    else:
        answer.append('no')

print(*answer)

📝코드 설명2

문제를 다 풀고 더 나은 풀이를 찾아보았는데, 가장 많이 나오는 풀이였습니다.
파이썬의 slice를 사용하면 더 간단하게 구현할 수 있습니다.

slice에서 각각의 항목은 [start : stop : step]를 의미하기 때문에, string[::-1]처럼 작성하면 반대 방향으로 리스트의 데이터를 자를 수 있습니다.

⏰ 메모리와 시간

reduce 사용

• 메모리 : 32408 KB
• 시간 : 92 ms

slice 사용

• 메모리 : 30840 KB
• 시간 : 64 ms

---

Reference

파이썬 reduce 함수 사용법