문제

SWEA 1206: [S/W 문제해결 기본] 1일차 - View https://swexpertacademy.com/main/code/problem/problemDetail.do?contestProbId=AV134DPqAA8CFAYh

풀이

• 현재 건물의 높이를 좌 우 2개의 건물 놀이와 비교한다.
• 좌 우 각각 2개의 건물의 최대 높이를 구하고 그 높이가 현재 건물의 높이보다 얼마나 낮은 지 구한다.

코드

import java.util.Scanner;
import java.io.FileInputStream;
 
class Solution
{
    static int N;
    static int[] arr; // 건물 넣는 배열
 
    public static void main(String[] args) throws Exception {
        Scanner sc = new Scanner(System.in);
 
        int T = 10; // TestCase 수
        for (int tc = 1; tc <= T; tc++) {
            N = sc.nextInt(); // 건물 갯수
            arr = new int[N];
            for (int i = 0; i < N; i++) {
                arr[i] = sc.nextInt();
            }
 
            int cnt = 0;
            for (int i = 2; i < N - 2; i++) {
                int max = Math.max(arr[i - 2], Math.max(arr[i - 1], Math.max(arr[i + 1], arr[i + 2])));
                if (arr[i] - max > 0) // 좌 우 건물 높이의 최댓값보다 현재 건물의 높이가 높으면
                    cnt += arr[i] - max; 
            }
 
            System.out.println("#" + tc + " " + cnt);
        } // end of TC
    } // end of Main
}

정리

• Math.max 메소드를 사용하면 쉽다.