프로그래머스 '리스트 자르기' Java Stream으로 구현 과정에서..

Question

/* * 정수 n과 정수 3개가 담긴 리스트 slicer 그리고 정수 여러 개가 담긴 리스트 num_list가 주어집니다. * slicer에 담긴 정수를 차례대로 a, b, c라고 할 때, n에 따라 다음과 같이 num_list를 슬라이싱 하려고 합니다. * * n &#61; 1 : num_list의 0번 인덱스부터 b번 인덱스까지 * n &#61; 2 : num_list의 a번 인덱스부터 마지막 인덱스까지 * n &#61; 3 : num_list의 a번 인덱스부터 b번 인덱스까지 * n &#61; 4 : num_list의 a번 인덱스부터 b번 인덱스까지 c 간격으로 * * a번 인덱스 &#61; slicer[0] * b번 인덱스 &#61; slicer[1] * c 간격 &#61; slicer[2] * * 올바르게 슬라이싱한 리스트를 return하도록 solution 함수를 완성해주세요. */ import java.util.*; import java.util.stream.*; class Solution { public int[] solution(int n, int[] slicer, int[] num_list) { int[] answer &#61; {}; switch(n) { case 1: answer &#61; Arrays.stream(num_list, 0, slicer[1] &#43; 1).toArray(); break; case 2: answer &#61; Arrays.stream(num_list, slicer[0], num_list.length).toArray(); break; case 3: answer &#61; Arrays.stream(num_list, slicer[0], slicer[1] &#43; 1).toArray(); break; case 4: answer &#61; Arrays.stream(num_list, slicer[0], slicer[1] &#43; 1).filter(x -> x % slicer[2] &#61;&#61; 0).toArray(); break; } return answer; } } n &#61; 4 일 떄의 case에서 문제가 발생합니다.num_list를 a 인덱스부터 b 인덱스까지 stream화 하는 부분까지는 성공했는데, 이후 filter 구간에서 문제가 발생하는 것을 확인했습니다.Stream의 구성 요소를 c 간격에 맞게 걸러내야 하는데, x가 Stream의 index가 아닌 Stream 개별 값이라서 그런 것 같은데.. 어떻게 소스를 고쳐야할까요?

RH.BY · Answer

방법을 찾았습니다. 애초에 num_list를 stream화 하는게 아니라 num_list의 index를 stream화 하여 필터를 적용하는데, 필터의 조건을 'index 값을 c 간격으로 나눴을 때 나머지가 0' 인 경우(c 간격에 해당인 경우)로 처리해 index 값들을 받아온 후 해당 index의 num_list 값을 배열화 하여 return하는 방법으로 처리했습니다.IntStream.range(slicer[0], slicer[1] + 1).filter(i -> (i - slicer[0]) % slicer[2] == 0).map(i -> num_list[i]).toArray();

으악 또 · Answer

스위치문은 함수랑 같이 안쓰면 그냥 이프문이 보기 더 좋아요아 함수네 ㅋㅈㅅbreak 빼고 리턴 박으면 더 좋은데return Arrays.stream(num_list, slicer[0], num_list.length).toArray();

프로그래머스 '리스트 자르기' Java Stream으로 구현 과정에서..

2개의 답변

#stream 목록