[테코테코1.5 1-12] 데이터 분석

💡테코테코 시즌 1.5 1회 모임 on-site 문제 풀이입니다. (2025.01.12)

 

문제

 

출처: 프로그래머스 - PCCE 기출문제 10. 데이터 분석

 

내용

 

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 함수를 완성해 주세요. 단, 조건을 만족하는 데이터는 항상 한 개 이상 존재합니다.

 

 

 

기록하기

💡 어디까지 생각해봤는지 단계적으로 기록해봅니다.

 

풀이

풀이 시간

시작 시각	종료 시각	총 소요 시간
11:24	11:40	16분

 

문제 분석

 

제약 사항 파악 & 테스트 케이스 작성

 

• 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에 따라 올바른 범위의 숫자로 주어집니다.
• 정렬 기준에 해당하는 값이 서로 같은 경우는 없습니다.

 

입력값 분석

💡 입력값을 분석하면 문제에서 요구하는 알고리즘의 시간 복잡도를 간접적으로 파악할 수 있습니다.

data	ext	val_ext	sort_by	result
[[1, 20300104, 100, 80], [2, 20300804, 847, 37], [3, 20300401, 10, 8]]	"date"	20300501	"remain"	[[3,20300401,10,8],[1,20300104,100,80]]

 

 

의사 코드 작성

💡 의사 코드는 동작 중심으로 작성하는 것이 중요합니다.

💡 의사 코드는 문제 해결 순서로 작성합니다.

💡 의사 코드를 충분히 테스트해봅니다.

solution(int[][] data, String ext, int val_ext, String sort_by) {
    // hashMap 생성
    // code:0, date:1, maximum:2, remain:3
    int filterIndex = map.get(ext);
    int sortByIndex = map.get(sort_by);
    // `data`에서 `ext` 값이 `val_ext`보다 작은 데이터만 뽑은 후, `sort_by`에 해당하는 값을 기준으로 오름차순으로 정렬
    return Arrays.stream(data)
        .filter(d -> d[filterIndex] < val_ext)
        .sorted(sortByIndex 로 오름차순);
}

 

 

 

구현

import java.util.HashMap;
import java.util.Arrays;

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

        int filterIndex = columns.get(ext);
        int sortByIndex = columns.get(sort_by);

        // `data`에서 `ext` 값이 `val_ext`보다 작은 데이터만 뽑은 후, `sort_by`에 해당하는 값을 기준으로 오름차순으로 정렬
        return Arrays.stream(data)
            .filter(d -> d[filterIndex] < val_ext)
            .sorted((d1, d2) -> Integer.compare(d1[sortByIndex], d2[sortByIndex]))
            .toArray(int[][]::new);
    }
}

• filter()API는 O(N)의 시간복잡도를 가진다.
• sort() API는 O(NlogN)의 시간복잡도를 가진다.

import java.util.HashMap;
import java.util.Arrays;
import java.util.Comparator;

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

        int filterIndex = columns.get(ext);
        int sortByIndex = columns.get(sort_by);

        // `data`에서 `ext` 값이 `val_ext`보다 작은 데이터만 뽑은 후, `sort_by`에 해당하는 값을 기준으로 오름차순으로 정렬
        return Arrays.stream(data)
            .filter(d -> d[filterIndex] < val_ext)
            .sorted(Comparator.comparingInt(arr -> arr[sortByIndex]))
            .toArray(int[][]::new);
    }
}

 

 

복기하기

• Arrays sort, List sort 리뷰 포인트 알아보기