[Python/BOJ] 1421. 나무꾼 이다솜

https://www.acmicpc.net/problem/1421

1421번: 나무꾼 이다솜

문제

이다솜은 나무꾼이다. 이다솜은 산신령이 준 금도끼와 은도끼를 이용해서 나무를 열심히 했다. 나무가 끝난 후에 나무들을 쳐다보면서 내가 왜 나무를 하면서 살까 생각하다가, 나무가 엄청나게 값어치가 있다는 것을 알고 나무를 팔러 시장에 가기로 했다.

지역 목재상에서 이다솜의 나무를 사려고 했지만, 너무 길이가 제멋대로여서 나무를 사는 것을 거절을 했다. 목재상의 조건은 일단 팔려고 하는 나무의 길이를 전부 같게 만들어 오라는 것이었다. (나무의 길이는 자연수로) 이다솜은 나무를 하나씩 여러 번 팔려고 했지만, 지역 목재상의 주인은 한 사람에게 평생 단 한번의 판매 기회를 제공하다고 했기 때문에, 이다솜은 근처 작업장으로 가서 나무를 자르기로 했다.

작업장에서 나무를 한 번 자를 때는, C원이 든다. 그리고, 지역 목재상에서 나무를 살 때는, 한 단위에 W원씩 준다. (다른 말로 하면, K개의 나무가 있고, 길이가 L이면, 이다솜은 K*L*W원을 벌 수 있다.)

이다솜이 현재 가지고 있는 나무의 길이가 주어졌을 때, 이다솜이 벌 수 있는 가장 큰 돈을 구하는 프로그램을 작성하시오.

입력

첫째 줄에 이다솜이 가지고 있는 나무의 개수 N과 나무를 자를 때 드는 비용 C와 나무 한 단위의 가격 W이 주어진다. 둘째 줄부터 총 N개의 줄에 이다솜이 가지고 잇는 나무의 길이가 한 줄에 하나씩 주어진다. N은 50보다 작거나 같은 자연수이고, C와 W는 10,000보다 작거나 같은 자연수이다. 그리고 나무의 길이는 모두 10,000보다 작거나 같은 자연수이다.

출력

첫째 줄에 이다솜이 벌 수 있는 돈의 최댓값을 출력한다.

풀이

Code

import sys
input = sys.stdin.readline

n, c, w = map(int, input().split())
trees = [int(input()) for _ in range(n)]
# 1. 최댓값 변수 선언
ans = -sys.maxsize
# 2.
for l in range(1, max(trees) + 1) :
    # 2-1. 자를 수 있는 나무 개수 세기
    k = profit = 0
    for tree in trees :
        k = tree // l
        # 이득 계산
        profit += max(0, k * l * w - k * c if tree % l != 0 else k * l * w - (k-1) * c)
    # 2-2. 최댓값 갱신
    ans = max(ans, profit)
# 3. 결과 출력
print(ans)