출처: https://school.programmers.co.kr/learn/courses/30/lessons/250121
문제 설명
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
본문의 내용과 동일합니다.
cpp를 응시하는 경우 리스트는 배열과 동일한 의미이니 풀이에 참고해주세요.
ex) 번호가 담긴 정수 리스트 numbers가 주어집니다. => 번호가 담긴 정수 배열 numbers가 주어집니다.
java를 응시하는 경우 리스트는 배열, 함수는 메소드와 동일한 의미이니 풀이에 참고해주세요.
ex) solution 함수가 올바르게 작동하도록 한 줄을 수정해 주세요. => solution 메소드가 올바르게 작동하도록 한 줄을 수정해 주세요.
내가 작성한 코드문
import java.util.*;
import java.util.Arrays;
import java.util.List;
class Solution {
public int[][] solution(int[][] data, String ext, int val_ext, String sort_by) {
int[][] answer = new int[data.length]data[0].length];
// 데이터는 ["코드 번호(code)", "제조일(date)", "최대 수량(maximum)", "현재 수량(remain)"]으로 구성
// index = 0 code, index = 1 date, index = 2 maximum, index = 3 remain
// 정수 리스트 data와 어떤 정보를 기준으로 데이터를 뽑아낼지를 의미하는 문자열 ext, 뽑아낼 정보의 기준값을 나타내는 정수 val_ext, 정보를 정렬할 기준이 되는 문자열 sort_by
// 현재 수량이 적은 순서로 정렬 가능
// data에서 ext 값이 val_ext보다 작은 데이터만 뽑은 후, sort_by에 해당하는 값을 기준으로 오름차순으로 정렬하여 return 하도록
int index1 = 0;
int index2 = 0;
for(int i = 0; i < data.length; i++){
for(int j = 0; j < data[i].length; j++){
if(data[i][1] < val_ext){// ext 값이 val_ext보다 작은을시에
answer[index1++][index2++] = data[i][j];// 그 데이터만 뽑은 후(answer배열에 저장)
}
}
}
List<int[]> list = Arrays.asList(answer);
Arrays.sort(list, (a, b) -> Integer.compare(a[1], b[1])); // sort_by에 해당하는 값을 기준으로 오름차순으로 정렬
// 그 다음에 다시 answer 배열로 변환
for(int i = 0; i < list.size(); i++){
answer[i] = list.get(i);
}
return answer;
}}
날짜(1번인덱스) ext가 val_ext 보다 작을때 그 값을
answer 배열에 데이터 저장및 오름차순 정렬하는것.
거기까지는 문제 파악까진 쉬웠는데
이거를 코드로 표현하는게 좀 어려웠다.
그리고 현재 코드의 문제점은..
- ext와 sort_by가 문자열인데, 이를 숫자 index로 변환하지 않음
→ 이건 무조건 "date" 기준으로 비교하는 것인데, ext가 "code"나 "remain"일 수도 있다.
→ 따라서 ext, sort_by 문자열을 index로 매핑해줘야 한다.
- answer 배열에 값을 넣는 방식이 잘못됨
answer[index1++][index2++] = data[i][j];
→ 이건 2차원 배열 전체를 한 칸에 잘못 채우는 방식이고, 원하는 "행 단위 복사"가 안 된다.
수정된 코드문.
import java.util.*;
class Solution {
public int[][] solution(int[][] data, String ext, int val_ext, String sort_by) {
// 1. ext와 sort_by를 index로 변환
Map<String, Integer> indexMap = new HashMap<>();
indexMap.put("code", 0);
indexMap.put("date", 1);
indexMap.put("maximum", 2);
indexMap.put("remain", 3);
int extIdx = indexMap.get(ext);
int sortIdx = indexMap.get(sort_by);
// 2. 조건에 맞는 행만 리스트에 담기
List<int[]> filtered = new ArrayList<>();
for (int[] row : data) {
if (row[extIdx] < val_ext) {
filtered.add(row);
}
}
// 3. 정렬
filtered.sort((a, b) -> Integer.compare(a[sortIdx], b[sortIdx]));
// 4. 리스트를 다시 배열로 변환
int[][] answer = new int[filtered.size()][data[0].length];
for (int i = 0; i < filtered.size(); i++) {
answer[i] = filtered.get(i);
}
return answer;
}}
index 매핑으로 ext, sort_by 모두 유동적으로 처리
List<int[]> 사용으로 유연하게 필터링
sort()에 람다식으로 원하는 열 기준 정렬
