[ 백준 / BOJ 16951 ] 블록 놀이 ( 자바 / JAVA )

[ 백준 / BOJ 16951 ] 블록 놀이 ( 자바 / JAVA )

728x90

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

문제

욱제는 높이가 1인 블록을 매우 많이 가지고 있고, 블록을 쌓아서 탑 N개를 만들었다. 탑은 일렬로 배열했고, 왼쪽에서부터 i번째 탑의 높이는 Ai이다.

욱제가 가장 좋아하는 정수는 K이다. 따라서, 인접한 두 탑의 높이 차이를 K로 만들려고 한다. 즉, Ai+1 - Ai = K를 만족해야 한다.

욱제가 1분 동안 할 수 있는 작업은 탑 하나를 고르고, 탑에 블록을 더 놓아서 높이를 크게 만드는 것 또는 탑에서 블록을 빼서 높이를 작게 만드는 것이다. 인접한 두 탑의 높이 차이를 K로 만드는데 필요한 시간을 구해보자. 욱제는 손이 매우 빠르기 때문에, 1분 동안 놓을 수 있는 블록 또는 뺄 수 있는 블록의 개수는 무한대이다.

탑의 높이는 항상 1보다 크거나 같아야 한다.

입력

첫째 줄에 탑의 수 N, 욱제가 가장 좋아하는 정수 K가 주어진다. 둘째 줄에는 탑의 높이 A1, A2, ..., AN이 주어진다.

출력

첫째 줄에 모든 인접한 두 탑의 높이 차이를 K로 만드는 최소 시간을 출력한다.

코드

import java.io.BufferedReader; import java.io.IOException; import java.io.InputStreamReader; import java.util.StringTokenizer; public class Main_BOJ_16951_블록놀이 { public static void main(String[] args) throws IOException { BufferedReader br = new BufferedReader(new InputStreamReader(System.in)); StringTokenizer stringTokenizer = new StringTokenizer(br.readLine()); int N = Integer.parseInt(stringTokenizer.nextToken()); int K = Integer.parseInt(stringTokenizer.nextToken()); int min = Integer.MAX_VALUE; int[] towers = new int[N]; stringTokenizer = new StringTokenizer(br.readLine()); for(int i = 0 ; i < N ; i++){ towers[i] = Integer.parseInt(stringTokenizer.nextToken()); } for(int i = 0 ; i < N ; i++){ int[] tmp = new int[N]; tmp[i] = towers[i]; boolean flag = false; for(int j = i-1; j >= 0 ; j--){ tmp[j] = tmp[j+1] - K; if(tmp[j] <= 0){ flag = true; break; } } if(flag) continue; for(int j = i+1; j < N ;j++){ tmp[j] = tmp[j-1] + K; } int cnt = 0; for(int j = 0 ; j < N ; j++){ if(towers[j] != tmp[j]) cnt++; } min = min < cnt ? min : cnt; } System.out.println(min); } }

728x90

from http://hyeyun.tistory.com/92 by ccl(A) rewrite - 2021-11-11 22:01:32