[Java] 큰 수 출력하기 | 인프런

문제

 

 

풀이

입력으로 받은 배열의 원소 중 원소 값이 자신의 바로 앞 원소 값보다 큰 원소를 출력하는 문제다. 예외상황이 있다면 첫 번째 원소는 앞 원소가 없기 때문에 무조건 출력을 해야 한다.

 

solution 함수 안에서 결과를 출력할 수도 있지만 ArrayList<Integer> 형으로 반환해서 메인 함수에서 출력하도록 코드를 작성했다.

알고리즘은 다음과 같다.

- 매개변수로 입력받은 배열의 첫 원소를 리스트에 담습니다.

- 반복문을 사용하여 두 번째 원소부터 마지막 원소까지 접근하여 앞 원소와 비교하여 값이 크면 리스트에 담습니다.

 

소스코드

import java.util.ArrayList;
import java.util.Scanner;

/*
* N개의 정수를 입력받아 자신의 앞 수보다 큰 수를 출력하세요.
* - 첫 번째 수는 무조건 출력합니다.
* - 비교할 앞의 수를 저장합니다.
* - 자신의 수와 앞의 수를 비교합니다.
* */
public class Main {
    public static ArrayList<Integer> solution(int n, int[] arr) {
        ArrayList<Integer> answer = new ArrayList<>();

        // 처음 수는 무조건 출력
        answer.add(arr[0]);

        // 두 번째 수부터 검사
        for (int i=1; i<arr.length; i++) {
            if (arr[i] > arr[i-1]) answer.add(arr[i]);
            arr[i-1] = arr[i];
        }

        return answer;
    }

    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        int n = in.nextInt();
        int[] arr = new int[n];
        for (int i=0; i<n; i++) {
            arr[i] = in.nextInt();
        }
        for (Integer x : solution(n, arr)) {
            System.out.print(x + " ");
        }
    }
}