[JAVA] 백준 2491 수열

엄청 쉬워보였는데 처음 로직을 잘못생각해서 삽질한 문제,,

원래 flag를 세워서 바꿔주려고 했는데, 반복하는 숫자가 나올때의 처리를 어떻게 해야할 지 모르겠어서 스터디원에게 물어보니까 간단한 로직을 알려줘서 그걸로 처음부터 다시 풀었다. ㅎ

그냥 반복문 두번 돌려주면 되는거였음 ㅎㅎ

그리고 처음에 또 틀렸습니다. 나와서 뭐지 하고 찾아보니까 입력 1개가 들어왔을때 내 코드는 0이 나오는게 문제였음. 그래서 N==1인 경우를 추가해줘서 해결했다.

 

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

public class bj2491 {
	public static void main(String[] args) throws NumberFormatException, IOException {
		BufferedReader in = new BufferedReader(new InputStreamReader(System.in));
		int N = Integer.parseInt(in.readLine());

		int[] num = new int[N];

		StringTokenizer st = new StringTokenizer(in.readLine(), " ");
		for (int i = 0; i < N; i++) {
			num[i] = Integer.parseInt(st.nextToken());
		}
		if (N == 1) {
			System.out.println(1);
			return;
		}

		int cntup = 1;
		int cntdown = 1;
		int upmax = 0;
		int downmax = 0;

		for (int i = 0; i < N - 1; i++) {
			if (num[i] <= num[i + 1]) {
				cntup++;
			} else {
				cntup = 1;
			}
			if (upmax < cntup) {
				upmax = cntup;
			}
		}

		for (int i = 0; i < N - 1; i++) {
			if (num[i] >= num[i + 1]) {
				cntdown++;
			} else {
				cntdown = 1;
			}
			if (downmax < cntdown) {
				downmax = cntdown;
			}
		}

		int max = Math.max(downmax, upmax);

		System.out.println(max);

	}
}