[알고리즘] 프로그래머스 1단계 - [PCCE 기출문제] 10번 / 데이터 분석

프로그래머스 문제 링크

https://school.programmers.co.kr/learn/courses/30/lessons/250121

 

문제 설명

AI 엔지니어인 현식이는 데이터를 분석하는 작업을 진행하고 있습니다. 데이터는 ["코드 번호(code)", "제조일(date)", "최대 수량(maximum)", "현재 수량(remain)"]으로 구성되어 있으며 현식이는 이 데이터들 중 조건을 만족하는 데이터만 뽑아서 정렬하려 합니다.

 

정렬한 데이터들이 담긴 이차원 정수 리스트 data와 어떤 정보를 기준으로 데이터를 뽑아낼지를 의미하는 문자열 ext, 뽑아낼 정보의 기준값을 나타내는 정수 val_ext, 정보를 정렬할 기준이 되는 문자열 sort_by가 주어집니다. data에서 ext 값이 val_ext보다 작은 데이터만 뽑은 후, sort_by에 해당하는 값을 기준으로 오름차순으로 정렬하여 return 하도록 solution 함수를 완성해 주세요. 단, 조건을 만족하는 데이터는 항상 한 개 이상 존재합니다.

 

예시

 

data

[[1, 20300104, 100, 80], [2, 20300804, 847, 37], [3, 20300401, 10, 8]]

 

ext  

"date" 

 

val_ext

20300501 

 

sort_by

"remain"

 

result

[[3,20300401,10,8],[1,20300104,100,80]]

 

import java.util.*;

class Solution {
    public int[][] solution(int[][] data, String ext, int val_ext, String sort_by) {
		// 기한 기준 변환
        int extStd = chooseStd(ext);
		// 분류 기준 변환
        int sortStd = chooseStd(sort_by);
		// 기한 기준으로 필터 처리한 후에 분류기준으로 정렬한다
        return Arrays.asList(data).stream().filter(item -> item[extStd] < val_ext)
                .sorted(Comparator.comparingInt(a -> a[sortStd]))
                .toArray(int[][]::new);
    }

	// 각 기준에 맞게 인덱스 번호 리턴
    public int chooseStd(String std) {

        return switch (std) {
            case "code" -> 0;
            case "date" -> 1;
            case "maximum" -> 2;
            case "remain" -> 3;
            default -> 4;
        };
    }
}