알고리즘/알고리즘 문제 풀이

(Python) 백준 1259번 - 팰린드롬수

용꿀 2022. 9. 23. 14:22

 

 

문제 링크 : https://www.acmicpc.net/problem/1259

 

1259번: 팰린드롬수

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

www.acmicpc.net

 

풀이

원래 브론즈 문제는 포스팅 하지 않으려 했으나 더 좋은 코드를 만드려고 노력하였던 기록을 남기고 싶어서 순서대로 풀이를 올려보겠다.

def cal(l):
    if len(l) == 1:
        return "yes"
    if len(l) == 2:
        if l[0] == l[1]:
            return "yes"
        return "no"
    if len(l) == 3:
        if l[0] == l[2]:
            return "yes"
        return "no"
    if len(l) == 4:
        if l[0] == l[3] and l[1] == l[2]:
            return "yes"
        return "no"
    if len(l) == 5:
        if l[0] == l[4] and l[1] == l[3]:
            return "yes"
        return "no"


while 1:
    l = input()
    if (int(l) == 0):
        break
    print(cal(list(l)))

 

처음에는 단순하게 입력받는 숫자의 길이대로 케이스를 나눠서 문제를 해결하였다. 

 

def cal(l):
    if len(l) == 1:
        return "yes"
    elif len(l) / 2 < 2:
        if l[0] == l[len(l)-1]:
            return "yes"
        return "no"
    else:
        if l[0] == l[len(l)-1] and l[1] == l[len(l)-2]:
            return "yes"
        return "no"


while 1:
    l = input()
    if (int(l) == 0):
        break
    print(cal(list(l)))

 

그 후에 간단한 규칙을 찾아서 숫자를 한 번 비교하는 경우와 두 번 비교하는 경우를 나눠 풀었다.

 

def cal(l):
    if len(l) == 1:
        return "yes"
    else:
        for i in range(0, len(l)//2):
            if l[i] != l[len(l)-(1+i)]:
                return "no"
        return "yes"

while 1:
    l = input()
    if (int(l) == 0):
        break
    print(cal(list(l)))

 

for문을 사용하여 더 간단하게 코드를 짤 수 있다고 생각하여 최종적으로 이렇게 간략화를 시켰다.