일기 대신 코드 슬쩍
[프로그래머스][JAVA] Lv1. 데이터 분석 본문
문제 설명
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에 따라 올바른 범위의 숫자로 주어집니다.
- 정렬 기준에 해당하는 값이 서로 같은 경우는 없습니다.
입출력 예
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]] |
입출력 예 설명
입출력 예 #1
- 본문의 내용과 동일합니다.
아이디어
Stream API 맨날 쓰려고 했는데 사실 귀찮아서 항상 안 써왔지만… 이번에 한 번 짚고 넘어가는게 좋을 것 같아서 stream API에 대해 살펴보았다.(나말고 지피티가해주긴했는데..)
Java의 Stream API는 Java 8에서 도입되었으며, 컬렉션(Collection)의 처리를 매우 간결하고 효율적으로 할 수 있게 해주는 기능입니다. 데이터 컬렉션을 추상화한 스트림(Stream)을 통해 데이터를 필터링, 변환, 정렬, 집계 등 다양한 연산을 수행할 수 있습니다. Stream API는 함수형 프로그래밍의 접근 방식을 사용하며, 이를 통해 코드의 가독성을 높이고 병렬 처리를 쉽게 구현할 수 있습니다.
주요 특징
선언적 처리: 데이터를 어떻게 처리할지(How)보다 무엇을 수행할 것인지(What)에 집중할 수 있게 해줍니다.불변성: 스트림 연산은 원본 데이터를 변경하지 않습니다. 연산 결과를 새로운 스트림으로 반환합니다.병렬 처리 지원: parallelStream() 메서드를 통해 데이터를 병렬로 처리할 수 있어, 멀티코어 프로세서의 이점을 살릴 수 있습니다.
주요 연산
Stream API는 중간 연산(intermediate operations)과 최종 연산(terminal operations)으로 나뉩니다.
중간 연산: 필터링(filter), 매핑(map), 정렬(sorted) 등 데이터 스트림을 변환하는 연산으로, 다른 스트림을 반환합니다. 중간 연산은 게으른 연산으로, 최종 연산이 호출되기 전까지는 실행되지 않습니다.
최종 연산: forEach, collect, reduce 등 스트림 처리를 마무리하고 결과를 반환하는 연산입니다. 최종 연산을 호출하면 그때서야 중간 연산이 모두 수행되고 스트림은 소비됩니다.
예시
다음은 Java의 Stream API를 사용하는 간단한 예시입니다.
javaimport java.util.Arrays;
import java.util.List;
import java.util.stream.Collectors;
public class StreamExample {
public static void main(String[] args) {
List<String> myList = Arrays.asList("apple", "banana", "cherry", "date");
// 문자열의 길이가 5보다 큰 요소만 필터링하고, 대문자로 변환한 후, 결과를 리스트로 수집
List<String> filteredList = myList.stream()
.filter(s -> s.length() > 5)
.map(String::toUpperCase)
.collect(Collectors.toList());
System.out.println(filteredList);// 출력: [BANANA, CHERRY]
}
}
이 예시에서는 문자열 리스트를 스트림으로 변환한 후, 길이가 5보다 큰 요소만 필터링하고, 모든 문자열을 대문자로 변환하여 새로운 리스트로 수집하는 과정을 보여줍니다.
문제 풀이
import java.util.Arrays;
import java.util.List;
import java.util.stream.Collectors;
public class Solution {
public static List<int[]> solution(int[][] data, String ext, int val_ext, String sort_by) {
// ext와 sort_by에 따라 index 결정
int extIndex = getIndex(ext);
int sortByIndex = getIndex(sort_by);
// 조건에 맞는 데이터 필터링
List<int[]> filteredData = Arrays.stream(data)
.filter(d -> d[extIndex] < val_ext)
.collect(Collectors.toList());
// sort_by에 해당하는 값으로 정렬
filteredData.sort((d1, d2) -> Integer.compare(d1[sortByIndex], d2[sortByIndex]));
return filteredData;
}
// ext와 sort_by 문자열을 인덱스 값으로 변환
private static int getIndex(String column) {
switch (column) {
case "code":
return 0;
case "date":
return 1;
case "maximum":
return 2;
case "remain":
return 3;
default:
throw new IllegalArgumentException("잘못.침.ㅋ");
}
}
}
'코딩테스트 > 프로그래머스(JAVA)' 카테고리의 다른 글
[프로그래머스][JAVA] Lv0. 숨어있는 숫자의 덧셈(1) (0) | 2024.05.25 |
---|---|
[프로그래머스][JAVA] Lv1. 성격 유형 검사하기 (0) | 2024.05.21 |
[프로그래머스][JAVA] Lv0. 숨어있는 숫자의 덧셈(1) (0) | 2024.05.21 |
[프로그래머스][JAVA] Lv1. 신규 아이디 추천 (0) | 2024.05.18 |
[프로그래머스][JAVA] Lv0. OX퀴즈 (0) | 2024.05.18 |