⏹️ [SW Expert Academy] [Python] [D3] 1206. View

문제

https://swexpertacademy.com/main/code/problem/problemDetail.do?contestProbId=AV134DPqAA8CFAYh

SW Expert Academy

풀이

SWEA의 D3 문제이다. 문제를 제대로 읽지 않아서 왼쪽 2칸과 오른쪽 두칸은 높이가 0인 것을 간과하였다. 문제를 해결하기 위해 무조건 모든 조건을 정확히 읽고 해결해야겠다. 왼쪽과 오른쪽 끝의 높이가 0이므로 문제의 난이도가 확 낮아진다. 단순히 자신의 위치에서 왼쪽과 오른쪽 2칸 안에 있는 모든 칸 중에 가장 높은 칸의 값을 빼면 조망권의 세대 수가 나오게 된다.  

T = int(input())
# 여러개의 테스트 케이스가 주어지므로, 각각을 처리합니다.
for test_case in range(1,11):
    result = 0
    houseCount = int(input())
    house = list(map(int , input().split()))
    for i in range(2, houseCount-2):
        arMax = max(house[i-1],house[i-2],house[i+1],house[i+2])
        if house[i] > arMax:
            result += ( house[i] - arMax )

    print("#{} {}".format(test_case,result))

SWEA는 처음이라 문제 제출 방식이나 테스트 케이스가 적응이 좀 안되는 것 같다. 백준이랑 프로그래머스에 비해 어렵다고 해야하나..? 조금 더 많은 문제들을 풀어보면서 백준이랑 난이도 비교를 해봐야 할 것 같다. 삼성 코딩테스트를 앞두고 더 많은 문제를 풀어봐야 한다. 

 

Java 코드

package SWEA;
import java.util.Scanner;

class swea_1204
{
    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);
        }
    }
}