BOJ 2491 수열 📈📉

문제 링크

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를 출력하여야 한다.

입력

첫째 줄에는 수열의 길이 N이 주어지고, 둘째 줄에는 N개의 숫자가 빈칸을 사이에 두고 주어진다. N은 1 이상 100,000 이하의 정수이다.

출력

첫째 줄에 가장 긴 길이를 출력한다.

구조화

상승 수열의 길이를 cnt에 저장, 감소 수열의 길이를 cnt_b에 저장

수가 같을 경우는 둘다++

오답 노트

max 갱신 부분을 반복문 마지막에 둬야 max가 갱신됨

이전에는 이전값<다음값 일 때 cnt_b와 max를 비교 후 갱신, 이전값>다음값 일 때 cnt와 max를 비교 후 갱신

이럴 경우 조건이 발동되지 않으면 값의 갱신이 일어나지 않음

소스 코드

import java.io.*;
import java.util.*;

public class Main {
    public static void main(String[] args) throws IOException {
        int max = 1;

        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int N = Integer.parseInt(br.readLine());

        int cnt = 1;
        int cnt_b = 1;
        StringTokenizer st = new StringTokenizer(br.readLine());
        int cur = Integer.parseInt(st.nextToken());
        for (int i = 1; i < N; i++) {
            int in = Integer.parseInt(st.nextToken());
            if (cur<in) {
                cnt++;
                cnt_b = 1;
            }
            else if (cur>in) {
                cnt_b++;
                cnt = 1;
            }
            else {
                cnt++;
                cnt_b++;
            }

            cur = in;
            max = Math.max(max, cnt_b);
            max = Math.max(max, cnt);
        }
        System.out.println(max);
    }
}