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

데이터추출 로직을 너무 기본 Java기능만 사용해서 한것같다. Stream으로 Filter나 forEach를 걸어서 해보면 좋을듯

그리고 코테에서 사용하는 이차원배열 정렬을,, 다 까먹었다. 객체정렬하는 Comparable도 있었던것같은데 기록을 한번 뒤져보면 좋을듯. 실무에서는 쿼리상에서 정렬을 마쳐서 불러오니까.. 잘 안쓰게 되는 기능인것같다.

 

->배열을 Stream을 통해서 사용하고 다시 toArray를 하면 좋아보인다. Stream은 필수인듯.

---

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

예를 들어 다음과 같이 데이터가 주어진다면

data = [[1, 20300104, 100, 80], [2, 20300804, 847, 37], [3, 20300401, 10, 8]]
이 데이터는 다음 표처럼 나타낼 수 있습니다.

code date maximum remain
1 20300104 100 80
2 20300804 847 37
3 20300401 10 8
주어진 데이터 중 "제조일이 20300501 이전인 물건들을 현재 수량이 적은 순서"로 정렬해야 한다면 조건에 맞게 가공된 데이터는 다음과 같습니다.

data = [[3,20300401,10,8],[1,20300104,100,80]]
정렬한 데이터들이 담긴 이차원 정수 리스트 data와 어떤 정보를 기준으로 데이터를 뽑아낼지를 의미하는 문자열 ext, 뽑아낼 정보의 기준값을 나타내는 정수 val_ext, 정보를 정렬할 기준이 되는 문자열 sort_by가 주어집니다.

data에서 ext 값이 val_ext보다 작은 데이터만 뽑은 후, sort_by에 해당하는 값을 기준으로 오름차순으로 정렬하여 return 하도록 solution 함수를 완성해 주세요. 단, 조건을 만족하는 데이터는 항상 한 개 이상 존재합니다.

 

제한사항

 

• 1 ≤ data의 길이 ≤ 500
• data[i]의 원소는 [코드 번호(code), 제조일(date), 최대 수량(maximum), 현재 수량(remain)] 형태입니다.
• 1 ≤ 코드 번호≤ 100,000
• 20000101 ≤ 제조일≤ 29991231
• data[i][1]은 yyyymmdd 형태의 값을 가지며, 올바른 날짜만 주어집니다. (yyyy : 연도, mm : 월, dd : 일)
• 1 ≤ 최대 수량≤ 10,000
• 1 ≤ 현재 수량≤ 최대 수량
• ext와 sort_by의 값은 다음 중 한 가지를 가집니다.
• "code", "date", "maximum", "remain"
• 순서대로 코드 번호, 제조일, 최대 수량, 현재 수량을 의미합니다.
• val_ext는 ext에 따라 올바른 범위의 숫자로 주어집니다.
• 정렬 기준에 해당하는 값이 서로 같은 경우는 없습니다.

---

내 코드

import java.util.*;

class Solution {
    public int[][] solution(int[][] data, String ext, int val_ext, String sort_by) {
        int[][] temp = new int[data.length][4];
        int column = getDataIndex(ext);
        int sort = getDataIndex(sort_by);
        int length = 0; //answer의 배열
        
        //data중 조건에 맞는값을 먼저 temp에 추려냄
        for(int i = 0; i < data.length; i++){
            if(data[i][column] < val_ext){
                for(int j = 0; j < 4; j++){
                    temp[length][j] = data[i][j];
                }
                length++;
            }
        }
       
        //
        int[][] answer = new int[length][4];
                                         
        //temp에서 진짜데이터만 추출
        int idx = 0;
        for(int i = 0; i < data.length; i++){
            if(temp[i][0] != 0){
                for(int j = 0; j < 4; j++){
                    answer[idx][j] = temp[i][j];
                }
                idx++;
            }
        }
        
        Arrays.sort(answer, new Comparator<int[]>(){
            @Override
            public int compare(int[] o1, int[] o2){
                return o1[sort] - o2[sort];
            }
        });

        return answer;
    }

    public int getDataIndex(String val){
        switch(val){
            case "code":
                return 0;
            case "date":
                return 1;
            case "maximum":
                return 2;
            case "remain":
                return 3;
            default : 
                return -1;
        }
    }
}

 

---

다른 사람의 풀이

import java.util.*;

class Solution {
    public int[][] solution(int[][] data, String ext, int val_ext, String sort_by) {

        String[] arr = {"code","date","maximum","remain"};
        List<String> columnList = Arrays.asList(arr);
        int extIdx = columnList.indexOf(ext);
        int sortIdx = columnList.indexOf(sort_by);
        int[][] answer = Arrays.stream(data).filter(o1 -> o1[extIdx] < val_ext)
            .sorted((o1 ,o2) -> o1[sortIdx]-o2[sortIdx]).toArray(int[][]::new);

        return answer;
    }
}