[BOJ] 2491 - 수열

개인적인 풀이일 뿐, 최적의 정답이 아님을 알려드립니다.

 

문제

www.acmicpc.net/problem/2491

2491번: 수열

난이도: 실버 2

사용언어: JAVA

 

풀이

복잡하게 생각하지 말고, 배열에서 증가하는 부분과 감소하는 부분을 별개로 반복문을 수행하면 쉽게 정답을 찾아낼 수 있습니다.

import java.io.*;
import java.util.StringTokenizer;
 
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];
        int i;
        int max = 0; // 지금까지의 증가 또는 감소하는 수열 중 가장 긴 부분의 길이 저장
        int cnt = 1;
        String s = br.readLine();
        StringTokenizer st = new StringTokenizer(s);
        if (N == 1) { // 배열의 길이가 1일 때 예외처리 : 1이면 최대 길이는 무조건 1
            bw.write("1");
            bw.flush();
            System.exit(0);
        }
        for (i = 0; i < N; i++) {
            arr[i] = Integer.parseInt(st.nextToken());
        }
 
        for (i = 1; i < N; i++) { // 증가하는 구간 찾기
            if (arr[i] >= arr[i - 1]) { // 바로 이전 값보다 크거나 같으면
                cnt++; // +1
                max = Math.max(max, cnt); // max보다 크다면 max 값 갱신
            } else {
                cnt = 1; // 바로 이전 값보다 작다면 cnt 초기화
            }
        }
 
        cnt = 1;
        for (i = 1; i < N; i++) { // 감소하는 구간 찾기
            if (arr[i] <= arr[i - 1]) { // 바로 이전 값보다 작거나 같으면
                cnt++; // +1
                max = Math.max(max, cnt); // max보다 크다면 max 값 갱신
            } else {
                cnt = 1; // 바로 이전 값보다 크다면 cnt 초기화
            }
        }
        bw.write(String.valueOf(max));
        bw.flush();
    }
}