[프로그래머스] PCCE 기출문제 10번/데이터 분석

 

import java.util.*;

class Solution {
    public List<int[]> solution(int[][] data, String ext, int val_ext, String sort_by) {
        List<int[]> list = new ArrayList<>();
        String dataArr[] = {"code", "date", "maximum", "remain"};
        int extIndex = 0;
        int sortIndex = 0;
        
        for (int i = 0; i < 4; i++) {
            if (ext.equals(dataArr[i])) {
                extIndex = i;
            }
            if (sort_by.equals(dataArr[i])) {
                sortIndex = i;
            }
        }
        final int _sort = sortIndex;
        
        for (int i = 0; i < data.length; i++) {
            if (data[i][extIndex] < val_ext) {
                list.add(data[i]);
            }
        }

        Collections.sort(list, (o1, o2) -> o1[_sort] - o2[_sort]);
        
        return list;
    }
}