[BOJ/백준] 5361 - 전투 드로이드 가격 (Python)

문제

문제링크

전투 드로이드의 부품의 가격은 다음과 같다.

블래스터 라이플	시각 센서	청각 센서	팔	다리
$350.34	$230.90	$190.55	$125.30	$180.90

각 부품의 필요한 개수가 주어질 때 소수점 둘째 자리까지 필요 비용을 구하라.

 

풀이

총 비용을 계산하는 price() 함수를 작성한다.

각 부품의 가격을 저장한 리스트를 생성하고

for문으로 입력받은 부품의 수와 해당 가격을 곱한 값을 새 리스트에 저장하고 이 리스트의 sum을 return한다.

이 함수에 입력받은 부품 수 리스트를 넣고 f-string으로 달러 표시와 함께 소수점 둘째 자리까지 출력한다.

 

코드

def price(parts):
  part_price= [350.34, 230.90, 190.55, 125.30, 180.90]
  tot=sum([part_price[i]*parts[i] for i in range(5)])
  return tot

n=int(input())
for i in range(n):
  parts=list(map(int, input().split()))
  print(f"${price(parts):.2f}")