PCCE 기출문제(Java) - 데이터 분석

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

프로그래머스

import java.util.List;
import java.util.Arrays;

class Solution {
    public int[][] solution(int[][] data, String ext, int val_ext, String sort_by) {
        int[][] answer = {};
        String[] col = {"code", "date", "maximum", "remain"};
        List<String> columnList = Arrays.asList(col);
        int extIdx = columnList.indexOf(ext);
        int sortIdx = columnList.indexOf(sort_by);
        
        answer = Arrays.stream(data)
                .filter(row -> row[extIdx] < val_ext)
                .sorted((row1, row2) -> row1[sortIdx] - row2[sortIdx])
                .toArray(int[][]::new);
        
        return answer;
    }
}