[Python] 백준(1000번문제) A+B

문제

두 정수 A와 B를 입력받은 다음, A+B를 출력하는 프로그램을 작성하시오.

풀이

A와 B를 입력 받아서 A+B를 출력하는 프로그램을 작성해야 한다.

우선 A와 B를 입력 받기 위해 input()을 사용한다.

A = input()
B = input()

이렇게 각각 입력 받을 수도 있지만 두 개를 한 번에 받기 위해 input()과 split()을 함깨 사용한다.

A, B = input().split()

이렇게 하면 공백을 기준으로 A와 B를 입력 받을 수 있다.

>>> A, B = input().split()
13 16
>>> A
13
>>> B
16

이제 이 둘을 더해줘야 하는데 무턱대고 print(A+B)를 입력하면 틀렸다는 결과가 나온다.

왜냐면 이 때 입력 받은 A와 B의 타입이 str이기 때문이다.

>>> type(A)
str

그래서 A와 B를 int로 바꿔준 후에 A+B를 진행해줘야 한다.

A, B = input().split()
print(int(A)+int(B))

이렇게 하면 정답이 나옴!

• map()을 이용해서 하는 방법

  위에 적힌 코드를 아주 간단하게 map()을 이용해 풀 수 있다.

  map() : 함수를 연결시켜 시퀀스의 항목을 인자로 전달한 결과를 반환하는 함수

  map(function, iterable, …)

  예를 들어, 입력 값에 1을 더해주는 addone함수를 map에 적용 시켜 보자.

  >>> def addone(n):
     	  return n+1
  
  >>> 1st = list(map(addone,[10,20,30,40]))
  >>> print(1st)
  [11,21,31,41]

  이렇게 [10, 20, 30, 40]에 addone함수가 적용돼 [11, 21, 31, 41]이 결과값으로 나오게 된다.

  이제 map을 이용해 A+B를 풀어보면

  A, B = map(int,input().split())
  print(A+B)

  정답이 나온다~!