알고리즘
[백준 1495/파이썬(Python)] 기타리스트(실버 1)
Rudy
2023. 1. 31. 12:57
https://www.acmicpc.net/problem/1495
1495번: 기타리스트
첫째 줄에 N, S, M이 주어진다. (1 ≤ N ≤ 50, 1 ≤ M ≤ 1,000, 0 ≤ S ≤ M) 둘째 줄에는 각 곡이 시작하기 전에 줄 수 있는 볼륨의 차이가 주어진다. 이 값은 1보다 크거나 같고, M보다 작거나 같다.
www.acmicpc.net
💜풀이
- 곡 개수(n, 행)만큼 최대 볼륨값(m, 열) 길이의 배열을 만든다.
- d[i][j]=i 번째 곡을 j만큼의 볼륨으로 연주할 수 있는지의 여부
- 첫 시작 부분, 즉, d[0][s]를 1로 설정한다.
- 이중 루프 돌면서 만약 이전 곡의 현재 볼륨 위치에 1로 체크가 되어 있으면 현재 곡도 볼륨을 조정해서 연주할 수 있다. 다시 말해, 이전 곡의 볼륨 위치에 설정되어 있는 볼륨을 조정해 저장할 수 있다.
- 설정한 볼륨 j+=v[i-1]가 범위 내라면 그 위치를 1로 체크한다.
- 루프 돌면서 가장 마지막 행의 값이 1인 것 중에서 인덱스가 최대인 곳을 찾아 출력한다.
💜코드
n,s,m=map(int,input().split())
v=list(map(int,input().split()))
dp=[[0]*(m+1) for _ in range(n+1)]
dp[0][s]=1
for i in range(1,n+1):
for j in range(m+1):
if dp[i-1][j]>0:
if 0<=j+v[i-1]<=m:
dp[i][j+v[i-1]]=1
if 0<=j-v[i-1]<=m:
dp[i][j-v[i-1]]=1
answer=-1
for i in range(m+1):
if dp[n][i]==1:
answer=max(answer,i)
print(answer)