[알고리즘] BOJ 1495번 - 기타리스트(python3)

* 알고리즘 너무 약해서 기초 문제 50개 목표로 푸는 중...

* 3시간 안에 답을 내지 못하면 답지를 보고 30분 내로 정답 판정을 받고, 3일 뒤 다시 풀어보기

---

Greedy

9 / 50

 

탐색

12 / 50

 

기초 동적 프로그래밍

10 / 50(NEW!)

---

투포인터

2 / 10

 

이분탐색

0 / 10

---

문제

1495번: 기타리스트

→ solved.ac 기준 실버 1

---

문제 해결 아이디어

예제 1과 같은 경우: 시작 볼륨이 5, 최대 볼륨이 10이며

각 볼륨의 차이는 5, 3, 7이다.

그럼 첫 번째 곡은 0 혹은 10,

두 번째 곡은 3 혹은 7이 가능하며, -3과 13은 범위를 벗어나서 불가능하다.

세 번째 곡은 10 혹은 0이 가능하며, -4 혹은 14는 범위를 벗어나서 불가능하다.

마지막 곡이 조절 가능한 볼륨은 10 혹은 0이므로 정답은 10이다.

→ 각 곡마다 가능한 볼륨을 저장하고 업데이트하자.

→ dp[i번째 곡][가능한 볼륨]으로 하여 차례로 업데이트.

---

구현

내 풀이

import sys
input = sys.stdin.readline

# 곡의 갯수: n, 시작 볼륨: s, 최대 볼륨: m
n, s, m = map(int, input().split())
# 각 곡이 시작하기 전에 줄 수 있는 볼륨의 차이
v = list(map(int, input().split()))

dp = [[False] * (m + 1) for _ in range(n + 1)]
dp[0][s] = True # 첫 번째 시작 볼륨

for i in range(n):
  for j in range(m+1):
    if dp[i][j]:
      for x in [j + v[i], j - v[i]]:
        if 0 <= x and x <= m:
          dp[i+1][x] = True

answer = -1
for i in range(m + 1):
  if dp[n][i]:
    answer = max(answer, i)

print(answer)

---

메모

• 걸린 시간: 50분
• dp라고 해서 너무 어렵게 생각하다가 돌아간 문제....

---