[java][boj][s3] 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 이하의 정수이다.

출력

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

풀이

/*
 * 입력
 * 1. 수열의 길이 N
 * 2. N개의 숫자가 빈칸을 두고 주어짐(0~9)
 * 출력
 * 1. 가장 긴 길이
 * 조건
 * 1. 연속해서 커지거나(같은 것 포함) 연속해서 작아지는(같은 것 포함)중 가장 길이가 긴 것을 찾아내어 길이를 출력
 * 
 * >> 큰 것과 작은 것 두 개로 나눠서 둘 중에서 큰 것 출력
 */

나누는 것만 생각하면 간단한 코드다 처음에는 그 생각을 못해서 어렵게 풀었다가... 나중에는 간단하게 풀어봤다 개수가 1일 때 0이 출력되는 걸 못 잡아줘서 여러번 틀렸다

코드

package im형필수문제;

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

public class s3_2491_수열 {

	public static void main(String[] args) throws Exception {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		
		int N = Integer.parseInt(br.readLine());

		int[] num = new int[N];
		int upCnt = 1;
		int downCnt = 1;
		int result = 0;

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

		for (int i = 0; i < N-1; i++) {		
			if (num[i] <= num[i + 1]) upCnt++; //연속해서 커지는 숫자 찾기
			else upCnt =1;
			result = Math.max(upCnt, result);
		}

		for (int i = 0; i < N-1; i++) {			
			if (num[i] >= num[i + 1]) downCnt++; //연속해서 작아지는 숫자 찾기
			else downCnt = 1;
			result = Math.max(downCnt, result);
		}
		System.out.println(result);
	}
}

결과

 

#자바 #java #boj #백준 #알고리즘