[codeup] 130. 나이 계산 1

a,b = map(int, input().split())
a = str(a)
present = 2012
if b == 1 or b == 2:
  if len(a) == 5:
    year = int(a[:1])+1900
    print(present - year + 1)
  elif len(a) == 6:
    year = int(a[:2])+1900
    print(present - year + 1)
  elif len(a) <= 4:
    year = 1900
    print(present - year + 1)
else:
  if len(a) == 5:
    year = int(a[:1])+2000
    print(present - year + 1)
  elif len(a) == 6:
    year = int(a[:2])+2000
    print(present - year + 1)
  elif len(a) <= 4:
    year = 2000
    print(present - year + 1)

처음 코드 짜고 수정하면서 답안지

 

birth, gender = input().split()
gender = int(gender)
if (gender == 1 or gender == 2):
    birth_year = int('19' + birth[0:2])
    print(2012 - birth_year + 1)
elif (gender == 3 or gender == 4):
    birth_year = int('20' + birth[0:2])
    print(2012 - birth_year + 1)

짧게 짜고 싶어서 찾아서 수정해 본 답안지