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

문제 설명

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

---

제한 사항

---

입출력 예

---

풀이

import java.util.Arrays;
import java.util.Map;

class Solution {
    public int[][] solution(int[][] data, String ext, int val_ext, String sort_by) {
        Map<String, Integer> extMap = Map.of("code", 0, 
                                             "date", 1, 
                                             "maximum", 2, 
                                             "remain", 3);
        
        return Arrays.stream(data)
            .filter(arr -> arr[extMap.get(ext)] < val_ext)
            .sorted((int[] arr1, int[] arr2) -> arr1[extMap.get(sort_by)] - arr2[extMap.get(sort_by)])
            .toArray(int[][]::new);
    }
}

---

후기

오랜만에 스트림을 사용하려니까 많이 잊어버린 것 같다... 다시 꾸준히 풀어야 겠다. Comparator와 함수형 인터페이스에 대해서 더 공부해야겠다.