[프로그래머스/JAVA] 데이터 분석

목차 (OPEN)

• 🔗 문제 바로가기
• 📌 문제 요약
• 💡 아이디어
• 👩🏻‍💻 최종 코드
• 🕑 시간 복잡도

 

 

 

🔗 문제 바로가기

https://school.programmers.co.kr/learn/courses/30/lessons/250121?language=java

프로그래머스

 

---

📌 문제 요약

 

 

🎯 조건에 맞는 데이터 필터링  후 정렬하기

• 주어진 data: 각 요소는 [code, date, maximum, remain]
• 필터링 조건: ext 필드가 val_ext 보다 작은 경우만 추출
• 정렬 기준: sort_by 필드를 기준으로 오름차순 정렬
• ext와 sort_by는 "code", "date", "maximum", "remain" 중 하나

 

---

💡 아이디어

1️⃣ 필드명을 인덱스로 변환하기

2차원 배열 data는 각 행이 [code, date, maximum, remain] 순서로 이루어진 배열이다.

 

따라서 아래와 같은 데이터가 있다고 하면,

int[][] data = {
    {1, 20300104, 100, 80},
    {2, 20300804, 847, 37},
    {3, 20300401, 10, 8}
};

 

각 열의 의미는 아래와 같다.

인덱스	이름	설명
0	code	코드 번호
1	date	제조일 (yyyymmdd)
2	maximum	최대 수량
3	remain	현재 남은 수량

 

그러나 문제에서 ext 와 sort_by 는 문자열로 필드명이 주어지는데,

이 문자열을 실제 배열에서 사용하기 위해서는 각 필드명을 배열의 인덱스로 변환해야 한다.

 

그러므로 아래와 같이 매핑을 만들어주면 된다.

Map<String, Integer> indexMap = new HashMap<>();
indexMap.put("code", 0);
indexMap.put("date", 1);
indexMap.put("maximum", 2);
indexMap.put("remain", 3);

 

이렇게 하면, 이제 문자열 "date"가 들어오면 indexMap.get("date")를 통해 1이라는 인덱스를 얻을 수 있게 된다.

 

즉, "code", "date" 등 문자열을 직접 비교하지 않고, 배열 인덱스로 빠르게 접근할 수 있는 것이다.

 

 

2️⃣ 필터링하기

다음으로는 조건에 맞는 데이터만 걸러내야 한다.

 

예를 들어, ext = "date" 이고, val_ext = 20300501 이라면,

제조일(date)이 20300501보다 작은 항목만 필터링해야 한다.

 

이를 위해서, 먼저 앞서 만든 indexMap을 활용하여 "date"에 해당하는 인덱스를 얻고,

int extIndex = indexMap.get(ext); // 예: "date" → 1

 

그 후, 데이터를 하나씩 순회하면서 조건을 만족하는 항목만 리스트에 담으면 된다.

for (int[] d : data) {
    if (d[extIndex] < val_ext) {
        filtered.add(d); // 조건을 만족하는 행만 저장
    }
}

 

즉, d[1] < 20300501 인 경우만 골라내는 과정이라고 보면 된다.

 

 

3️⃣ 정렬하기

마지막으로는, 필터링된 데이터가 들어있는 리스트 filtered 를 정렬해주어야 한다.

 

정렬 기준은 sort_by 라는 문자열로 주어지기 때문에

이 역시 인덱스로 바꿔서 사용해야 한다.

int sortIndex = indexMap.get(sort_by); // 예: "remain" → 3

 

정렬 기준을 인덱스로 변환했다면,

아래와 같이 Java의 Comparator를 사용해서 정렬해주기만 하면 된다.

filtered.sort(Comparator.comparingInt(o -> o[sortIndex]));

 

이 코드는 다음과 같이 작동한다.

• o는 필터링된 각 행(배열)이다.
• o[sortIndex]는 정렬 기준에 해당하는 값이다.
• 예를 들어 sort_by = "reamin" 이라면 👉🏻 o[3] 즉, 남은 수량을 기준으로 정렬한다.

 

이렇게 하면 조건에 맞는 데이터만 모은 다음, 정렬 기준에 따라 오름차순으로 정리할 수 있다.

 

---

👩🏻‍💻 최종 코드

import java.util.*;

class Solution {
    public int[][] solution(int[][] data, String ext, int val_ext, String sort_by) {
        Map<String , Integer> indexMap = new HashMap<>();
        indexMap.put("code", 0);
        indexMap.put("date", 1);
        indexMap.put("maximum", 2);
        indexMap.put("remain", 3);
        
        int extIndex = indexMap.get(ext);
        int sortIndex = indexMap.get(sort_by);
        
        List<int[]> filtered = new ArrayList<>();
        for(int[] d : data){
            if(d[extIndex] < val_ext){
                filtered.add(d);
            }
        }
        
        filtered.sort(Comparator.comparingInt(o -> o[sortIndex]));
        
        int[][] answer = new int[filtered.size()][];
        for(int i = 0; i < filtered.size(); i++){
            answer[i] = filtered.get(i);
        }
        
        return answer;
    }
}

 

---

🕑 시간 복잡도

1️⃣ 필드 인덱스 매핑

Map<String, Integer> indexMap = new HashMap<>();

 

이 부분은 고정된 크기의 데이터를 다루는 것이므로 시간 복잡도는 O(1)이다.

 

 

2️⃣ 필터링 작업

for (int[] d : data) {
    if (d[extIndex] < val_ext) {
        filtered.add(d);
    }
}

 

data 배열을 한 번 순회하며 조건 검사를 하기 때문에 시간 복잡도는 O(N)이다.

 

 

3️⃣ 정렬 작업

filtered.sort(Comparator.comparingInt(o -> o[sortIndex]));

 

정렬 알고리즘은 Java 내부에서 TimSort를 사용하기 때문에 시간 복잡도는 O(M log M)이다.

이때, M은 필터링된 데이터의 개수이다.

 

최악의 경우에는 모든 행이 조건을 만족하므로, M = N이 될 수 있다.

따라서 이 경우 시간 복잡도는 O(N log N)이다.

 

 

4️⃣ 배열로 변환 작업

int[][] answer = new int[filtered.size()][];
for (int i = 0; i < filtered.size(); i++) {
    answer[i] = filtered.get(i);
}

 

리스트에서 배열로 복사하는 작업의 시간 복잡도는 O(N)이다.

 

 

👉🏻 따라서 최종 시간 복잡도는 O(N log N)이다.

      (전체 과정 중에서 가장 느린 작업인 정렬 작업이 시간 복잡도의 지배 항이 되기 때문이다.)