백준 25304번 영수증 - map()의 사용에 주의하라

백준 25304번 / 영수증 : https://www.acmicpc.net/problem/25304

 

이 문제는 구매한 물건들의 총 금액과 영수증에 적힌 총 금액이 일치하는지 검사하는 프로그램을 만드는 것이다. 영수증에는 구매한 각 물건의 가격과 개수, 그리고 구매한 물건들의 총 금액이 적혀있다.

 

이 문제를 풀어내는 과정은 다음과 같다.

 

• 영수증에 적힌 총 금액 X를 입력받는다.
• 구매한 물건의 종류의 수 N을 입력받는다.
• N번 반복하면서 각 물건의 가격 a와 개수 b를 입력받는다.
• 물건의 가격과 개수를 곱한 값(a * b)을 계속해서 누적하여 합계(실제 총 금액)를 구한다.
• 반복이 끝나면, 계산된 실제 총 금액과 영수증에 적힌 총 금액 X를 비교한다.
• 일치하면 "Yes"를, 일치하지 않으면 "No"를 출력한다.

내가 만든 코드는 다음과 같다.

X = int(input())
N = int(input())
sum = 0

for i in range(N):
    a,b = map(int,input().split())
    sum += a*b

if sum == X:
    print('Yes')
else:
    print('No')

 

 

map() 함수의 이용은 알아두면 정말 유용한 것 같다.