백준 - 2491번(수열)

문제 출처: https://www.acmicpc.net/problem/2491

문제

• 0에서부터 9까지의 숫자로 이루어진 N개의 숫자가 나열된 수열이 있다. 그 수열 안에서 연속해서 커지거나(같은 것 포함), 혹은 연속해서 작아지는(같은 것 포함) 수열 중 가장 길이가 긴 것을 찾아내어 그 길이를 출력하는 프로그램을 작성하라.

• 예를 들어 수열 1, 2, 2, 4, 4, 5, 7, 7, 2 의 경우에는 1 ≤ 2 ≤ 2 ≤ 4 ≤ 4 ≤ 5 ≤ 7 ≤ 7 이 가장 긴 구간이 되므로 그 길이 8을 출력한다. 수열 4, 1, 3, 3, 2, 2, 9, 2, 3 의 경우에는 3 ≥ 3 ≥ 2 ≥ 2 가 가장 긴 구간이 되므로 그 길이 4를 출력한다. 또 1, 5, 3, 6, 4, 7, 1, 3, 2, 9, 5 의 경우에는 연속해서 커지거나 작아지는 수열의 길이가 3 이상인 경우가 없으므로 2를 출력하여야 한다.

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.StringTokenizer;

public class Main {

    public static void main(String[] args) throws IOException {
        BufferedReader reader = new BufferedReader(new InputStreamReader(System.in));
        int N = Integer.parseInt(reader.readLine());
        int[] arr = new int[N];
        StringTokenizer tokenizer = new StringTokenizer(reader.readLine());

        int result = 1;
        int ascCount = 1;
        int descCount = 1;

        for (int i = 0; i < N; i++) {
            arr[i] = Integer.parseInt(tokenizer.nextToken());
        }

        int temp = arr[0];

        for (int i = 1; i < N; i++) {
            if (arr[i] > temp) {
                ascCount++;
                result = Math.max(descCount, result);
                descCount = 1;
            } else if (arr[i] < temp) {
                descCount++;
                result = Math.max(ascCount, result);
                ascCount = 1;
            } else {
                ascCount++;
                descCount++;
            }

            temp = arr[i];
        }

        result = Math.max(Math.max(ascCount, descCount), result);
        System.out.println(result);
    }

}

• 해결방법을 구상하기까지 시간이 꽤 걸렸다.
• 주요 로직은 단순히 2번 탐색한다. 한번은 커지는 것을 카운트하고, 한번은 작아지는 것을 카운트한다.
• 이를 위해 처음에는 커지는 것과 작아지는 탐색 반복을 나눠서 처리했으나, 코드를 완성하고 나니 한번의 반복으로 가능함을 보아 하나로 합쳐서 풀이하였다.