[Hackerrank] Week1 - Time Conversion

문제

AM, PM 시간을 입력 받아 24시 기준으로 변환하는 문제

코드

def timeConversion(s):
    if "PM" in s:
        if s[:2] == '12':
            return s[:-2]
        else:
            s = s.replace(s[:2], str(int(s[:2]) + 12))
            return s[:-2]

    elif "AM" in s:
        if s[:2] == '12':
            s = s.replace(s[:2], "00")
            return s[:-2]
        else:
            return s[:-2]