[Codeforces] 1005A: Tanya and Stairways

https://codeforces.com/problemset/problem/1005/A

Problem - 1005A - Codeforces

코드

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

public class Main { 

	public static void main(String[] args) throws IOException {
		BufferedReader bf = new BufferedReader(new InputStreamReader(System.in));
		int n = Integer.parseInt(bf.readLine());
		StringTokenizer st = new StringTokenizer(bf.readLine());
		int[] arr = new int[n];
		for(int i=0; i<arr.length; i++)
			arr[i] = Integer.parseInt(st.nextToken());
		
		int stairways = 1;	// 계단 수
		List<Integer> list = new ArrayList<Integer>();	// 각 계단에 포함된 계단수
		for(int i=0; i<arr.length-1; i++) {
			if(arr[i] >= arr[i+1]) {	// 계단 수가 감소하는 부분 -> 계단 수 증가
				stairways ++;
				list.add(arr[i]);
			}
		}
		list.add(arr[arr.length-1]);	// 마지막 계단은 무조건 포함
		
		System.out.println(stairways);
		for(int i : list)
			System.out.print(i + " ");
		
		bf.close();
	}

}

풀이

주어진 예제에서 계단수와, 각 계단의 수를 출력하는 문제다.

위에선 [1, 2, 3]의 세 개짜리 계단과 [1, 2, 3, 4]의 네 개짜리 계단 두개가 있다.

따라서 총 2개의 계단과 3칸, 4칸 계단을 출력한다.

 

먼저 입력을 받고, 배열에 저장한다.

arr= [1, 2, 3, 1, 2, 3, 4]

그 후 배열의 값 중 이전 값보다 작아지는 경우[3, 1]

새로운 계단의 시작이므로 계단수를 ++시켜주고, 작아지기 전의 값[3]을 ArrayList에 저장한다.

 

위 과정을 배열이 끝날때까지 반복하는데, 배열의 마지막 값은 항상 가장 큰 값이므로 ArrayList에 저장한다.

그 후 계단수와 각 계단이 몇개짜리 계단인지 출력하면 된다.