알고리즘) 프로그래머스 PCCE 기출문제 1회 10번(해시 사용)

해시를 사용하면 공간 복잡도를 많이 줄일 수 있는 간단한 문제입니다.

class Solution {
	public int[][] solution(int[][] data, String ext, int val_ext, String sort_by) {
        Map<String, Integer> fields = new HashMap<>();
        fields.put("code", 0);
        fields.put("date", 1);
        fields.put("maximum", 2);
        fields.put("remain", 3);

        int[][] list = new int[data.length][fields.size()];

        int j = 0;
        for (int i = 0; i < data.length; i++) {
            int val = data[i][fields.get(ext)];
            int sortBy = data[i][fields.get(sort_by)];

            if (val < val_ext) {
                list[j] = data[i];
                j++;
            }
        }

        Arrays.sort(list, (e1, e2) -> e1[fields.get(sort_by)] - e2[fields.get(sort_by)]);

        int k = 0;
        int[][] answer = new int[j][4];
        for (int i = 0; i < list.length; i++) {
            if (list[i][0] != 0) {
                answer[k++] = list[i];
            }
        }
        return answer;
    }
}