-
[Java/자바][백준 2156: 포도주 시식]아가개발자/자료구조,알고리즘 2021. 8. 17. 11:23
https://www.acmicpc.net/problem/2156
Problem
- 효주는 포도주 시식회에 갔다. 그 곳에 갔더니, 테이블 위에 다양한 포도주가 들어있는 포도주 잔이 일렬로 놓여 있었다. 효주는 포도주 시식을 하려고 하는데, 여기에는 다음과 같은 두 가지 규칙이 있다.
- 포도주 잔을 선택하면 그 잔에 들어있는 포도주는 모두 마셔야 하고, 마신 후에는 원래 위치에 다시 놓아야 한다.
- 연속으로 놓여 있는 3잔을 모두 마실 수는 없다.
예를 들어 6개의 포도주 잔이 있고, 각각의 잔에 순서대로 6, 10, 13, 9, 8, 1 만큼의 포도주가 들어 있을 때, 첫 번째, 두 번째, 네 번째, 다섯 번째 포도주 잔을 선택하면 총 포도주 양이 33으로 최대로 마실 수 있다.
Input
첫째 줄에 포도주 잔의 개수 n이 주어진다. (1≤n≤10,000) 둘째 줄부터 n+1번째 줄까지 포도주 잔에 들어있는 포도주의 양이 순서대로 주어진다. 포도주의 양은 1,000 이하의 음이 아닌 정수이다.
Output
첫째 줄에 최대로 마실 수 있는 포도주의 양을 출력한다.
Insight
1. DP(동적 계획법)을 통해서 문제를 해결할 수 있다.
2. 3잔을 연속해서 마실 수 없다.
- N개의 잔이 있을 때 N번째 잔을 안 마실 경우 N-1번째 잔을 마셨을 수도 있고, 안 마셨을 수도 있다. = N-1번째 잔을 마신 경우와 안 마신 경우 중에 최대값- N번째 잔을 마실 경우 N-1번째 잔과 N-2번째 잔 중 하나만 마실 수 있다 = N-2번째 잔을 마시고 N-1번째 잔을 안 마신 경우와 N-1번째 잔을 마시고 N-2번째 잔을 안 마신 경우 중 최대 값 + N번째 잔의 값
3. Memoization배열은 d[N][2]로 구성
d[N][0] = N번째 잔을 안 마실 때의 최대값
d[N][1] = N번째 잔을 마실 때의 최대값
import java.util.*; import java.io.*; public class Main { public static void main(String[] args) throws IOException{ BufferedReader br = new BufferedReader(new InputStreamReader(System.in)); BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out)); int N = Integer.parseInt(br.readLine()); int[] arr = new int[N+1]; int[][] d = new int[N+1][2]; for(int i=1; i<=N; i++) { arr[i] = Integer.parseInt(br.readLine()); } for(int i=1; i<=N; i++) { if(i==1) { d[i][0] = 0; d[i][1] = arr[i]; } else { d[i][0]= Math.max(d[i-1][0], d[i-1][1]); d[i][1] = Math.max(d[i-2][0]+arr[i-1], d[i-1][0])+arr[i]; } } int res = Math.max(d[N][0], d[N][1]); bw.write(String.valueOf(res)+"\n"); bw.flush(); bw.close(); } }
'아가개발자 > 자료구조,알고리즘' 카테고리의 다른 글
[JAVA/자바 - 그래프] 인접 행렬 구현 (0) 2021.09.15 [Java/자바][백준 1932: 정수 삼각형] (0) 2021.08.17 [Java/자바][백준 1149: RGB 거리] (0) 2021.08.16 [Java/자바][백준 2579: 계단 오르기] (0) 2021.08.12 [Java/자바][백준 9095: 1, 2, 3 더하기] (0) 2021.08.10 - 효주는 포도주 시식회에 갔다. 그 곳에 갔더니, 테이블 위에 다양한 포도주가 들어있는 포도주 잔이 일렬로 놓여 있었다. 효주는 포도주 시식을 하려고 하는데, 여기에는 다음과 같은 두 가지 규칙이 있다.