[SWEA-1284][Python] 수도 요금 경쟁

문제

삼성전자에 입사한 종민이는 회사 근처로 이사를 하게 되었다.

그런데 집의 위치가 두 수도 회사 A, B 중간에 위치하기에 원하는 수도 회사를 선택할 수 있게 되었는데, 두 회사 중 더 적게 수도 요금을 부담해도 되는 회사를 고르려고 한다.
 
종민이가 알아본 결과 두 회사의 수도 요금은 한 달 동안 사용한 수도의 양에 따라 다음과 같이 정해진다.
 
A사 : 1리터당 P원의 돈을 내야 한다.

B사 : 기본 요금이 Q원이고, 월간 사용량이 R리터 이하인 경우 요금은 기본 요금만 청구된다. 하지만 R 리터보다 많은 양을 사용한 경우 초과량에 대해 1리터당 S원의 요금을 더 내야 한다.
 

종민이의 집에서 한 달간 사용하는 수도의 양이 W리터라고 할 때, 요금이 더 저렴한 회사를 골라 그 요금을 출력하는 프로그램을 작성하라.

 

입력

첫 번째 줄에 테스트 케이스의 수 T가 주어진다.

각 테스트 케이스마다 첫 번째 줄에 위 본문에서 설명한 대로 P, Q, R, S, W(1 ≤ P, Q, R, S, W ≤ 10000, 자연수)가 순서대로 공백 하나로 구분되어 주어진다.

 

출력

각 테스트 케이스마다 ‘#x’(x는 테스트케이스 번호를 의미하며 1부터 시작한다)를 출력하고, 종민이가 내야 하는 수도 요금을 출력한다.

 

예제

[입력]	[출력]
2 9 100 20 3 10 8 300 100 10 250	#1 90 #2 1800

 

문제 해결 방법

• 문제의 설명을 이해하고 삼항 연산자를 이용하면 쉽게 풀 수 있었던 문제였다.

P, Q, R, S, W = map(int, input().split())

fare_A = P * W
fare_B = Q if W <= R else Q + (S * (W - R))

answer = fare_A if fare_A <= fare_B else fare_B

 

코드

T = int(input())

for test_case in range(1, 1 + T):
    P, Q, R, S, W = map(int, input().split())

    # P : 1L당 요금 (A사)
    # Q : RL 이하 요금 (B사)
    # S : RL 보다 많을 경우 초과량에 대해 1L당 요금
    # W : 한 달 동안 사용하는 수도의 양

    fare_A = P * W
    fare_B = Q if W <= R else Q + (S * (W - R))

    answer = fare_A if fare_A <= fare_B else fare_B

    print(f"#{test_case} {answer}")

 

참고

• 난이도: D2