13302번: 리조트
수영이는 여름방학을 맞이하여 많은 놀이 시설이 있는 KOI 리조트에 놀러가려고 한다. 리조트의 하루 이용권의 가격은 만원이다. 하지만 리조트의 규모는 상상을 초월하여 모든 시설을 충분히
www.acmicpc.net
문제
수영이는 여름방학을 맞이하여 많은 놀이 시설이 있는 KOI 리조트에 놀러가려고 한다. 리조트의 하루 이용권의 가격은 만원이다. 하지만 리조트의 규모는 상상을 초월하여 모든 시설을 충분히 즐기기 위해서는 하루로는 터무니없이 부족하다. 그래서 많은 이용객들은 3일 이상 연속으로 이용하기도 한다. KOI 리조트에서는 3일 연속 이용권을 할인된 가격 이만오천원에, 연속 5일권은 삼만칠천원에 판매하고 있다. 게다가 연속 3일권, 연속 5일권에는 쿠폰이 각각 1장, 2장이 함께 포함되어 있다. 쿠폰 3장은 하루 이용권 한 장으로 교환할 수 있다.
이용권 종류가격쿠폰지급
하루 이용권 | 10,000원 | 없음 |
연속 3일권 | 25,000원 | 쿠폰 1장 |
연속 5일권 | 37,000원 | 쿠폰 2장 |
연속 3일권과 연속 5일권은 구입일로부터 연속으로 3일 혹은 5일간만 이용이 가능하지만 해당 기간을 모두 이용할 필요는 없다.
수영이는 N일의 여름방학 중 다른 일정으로 리조트에 갈 수 없는 날이 M일 있다. KOI 리조트를 사랑하는 수영이는 그 외의 모든 날을 KOI 리조트에서 보내고자 한다. 물론, 가장 저렴한 비용으로 리조트를 이용하고자 한다.
예를 들어, 여름방학이 13일이라고 하고, 여름방학 기간 중 리조트에 갈 수 없는 날이 4번째, 6번째, 7번째, 11번째, 12번째 날이라고 하자. 다음 표의 첫 번째 행은 13일의 여름방학을 나타내고, 리조트에 갈 수 없는 날은 검정색으로 표시되어 있다. 표의 두 번째 행과 세 번째 행은 수영이가 이용권을 구입하는 두 가지 방법을 나타낸다.
두 번째 행의 구입 방법은 다음과 같다. 여름방학의 첫 번째 날에 연속 3일권을 구입하여 3번째 날까지 리조트를 이용하고, 구매시 1장의 쿠폰을 받는다. 5번째 날에는 하루 이용권을 구입하여 이용한다. 8번째 날에는 연속 3일권을 구입하여 10번째 날까지 리조트를 이용하고, 역시 구매시 쿠폰 1장을 받는다. 13번째 날에는 하루 이용권을 구입하여 리조트를 이용한다. 이렇게 하여 수영이가 리조트 이용을 위해 지불한 전체 비용은 70,000원이다.
세 번째 행은 더 저렴한 비용으로 리조트를 이용하는 구입 방법이다. 여름방학의 첫 번째 날에 연속 5일권을 구입하여 5번째 날까지 리조트를 이용하고(4번째 날 제외), 구매시 2장의 쿠폰을 받는다. 그리고 8번째 날에 연속 3일권을 구입하여 10번째 날까지 리조트를 이용하고, 역시 구매시 쿠폰 1장을 받는다. 13번째 날에는 그때까지 받은 3장의 쿠폰을 하루 이용권 한 장으로 교환하여 리조트를 이용한다. 이렇게 하여 수영이가 리조트 이용을 위해 지불한 전체 비용은 62,000원이다.
여름방학 기간과 리조트에 갈 수 없는 날의 정보가 주어질 때, 리조트를 이용하기 위해서 수영이가 지불해야 하는 최소비용을 계산하는 프로그램을 작성하시오.
입력
표준 입력으로 다음 정보가 주어진다. 첫 번째 줄에는 수영이의 여름방학의 일수를 뜻하는 정수 N(1 ≤ N ≤ 100)과 수영이가 리조트에 갈 수 없는 날의 수 M (0 ≤ M ≤ N)이 순서대로 주어진다. M이 0인 경우 더 이상의 입력은 주어지지 않으며, M이 0보다 큰 경우 그 다음 줄에는 수영이가 리조트에 갈 수 없는 날이 1 이상 N 이하의 정수로 날짜 순서대로 M개 주어진다.
예를 들어, M이 3이고 입력의 두 번째 줄에 정수 “12 14 17”이 주어진다면 여름방학의 12번째, 14번째, 17번째 날에는 리조트에 갈 수 없음을 의미한다.
출력
표준 출력으로 주어진 입력에서 제시된 날들을 제외한 나머지 날 모두 리조트에 입장하기 위해서 지불해야 하는 비용의 최솟값을 출력한다.
서브태스크
번호 | 배점 | 제한 |
1 | 11 | N ≤ 5 |
2 | 17 | M = 0 |
3 | 72 | 원래의 제약조건 이외에 아무 제약조건이 없다. |
코드
n, m = map(int, input().split())
visit = [True] * (n+2)
if m > 0:
for d in list(map(int, input().split())):
visit[d] = False
dp = [[1001]*50 for _ in range(n+6)] # 행: 날짜, 열: 쿠폰 수
dp[0][0] = 0
for i in range(n+1):
for j in range(40):
if dp[i][j] == 1001:
continue
if not visit[i+1]: # 내일은 안 가는 날
dp[i+1][j] = min(dp[i][j], dp[i+1][j])
continue
if j >= 3: # 쿠폰 사용
dp[i+1][j-3] = min(dp[i][j], dp[i+1][j-3])
# 1일권
dp[i+1][j] = min(dp[i][j]+10, dp[i+1][j])
# 3일권
for k in range(1, 4):
dp[i+k][j+1] = min(dp[i+k][j+1], dp[i][j]+25)
# 5일권
for k in range(1, 6):
dp[i+k][j+2] = min(dp[i+k][j+2], dp[i][j]+37)
print(min(dp[n])*1000)
행은 날짜, 열은 쿠폰 개수인 2차원 배열 dp를 만들어서 다이나믹 프로그래밍으로 해결하였다.
dp배열을 만들 때는 열은 가능한 쿠폰의 개수 이상인 50으로 하고, 마지막에 5일권을 사는 경우까지 고려하기 위해 행은 입력된 날짜 n+5행까지 만들어지도록 한다.
DP
- 만약 dp[i][j]라면 i번째 날에 j개의 쿠폰을 가질 수 있는 경우는 없다는 뜻이므로 continue한다.
- i+1일, 즉 내일이 안 가는 날이라면 dp[i+1][j]의 값은 dp[i][j]와 dp[i+1][j] 중 최소값으로 갱신만 해주고 continue한다.
- 만약 쿠폰이 3개 이상이라 사용할 수 있는 경우에는 dp[i+1][j-3]의 값을 쿠폰을 사용했을 때의 값이 더 작다면 바꿔준다.
- 그리고 1일권, 3일권, 5일권을 사용하는 경우를 고려해주는데, 3일권, 5일권의 경우에는 3일 뒤의 값만 바꿔주면 안되고, 해당되는 기간은 모두 값을 최소값으로 갱신해줘야 한다.
'문제풀이 > DP' 카테고리의 다른 글
[Python/파이썬] 백준 4095번 최대 정사각형 (0) | 2024.02.10 |
---|---|
[Python/파이썬] 백준 17484번 진우의 달 여행 (Small) (0) | 2024.02.04 |
[Python/파이썬] 백준 1958번 LCS 3 (0) | 2024.01.05 |
[Python/파이썬] 백준 11060번 점프 점프 (0) | 2023.12.25 |
[Python/파이썬] 백준 1309번 동물원 (1) | 2023.12.02 |