[프로그래머스 코딩기초 트레이닝] 리스트 자르기 (JAVA)

문제 설명

 

정수 n과 정수 3개가 담긴 리스트 slicer 그리고 정수 여러 개가 담긴 리스트 num_list가 주어집니다. slicer에 담긴 정수를 차례대로 a, b, c라고 할 때, n에 따라 다음과 같이 num_list를 슬라이싱 하려고 합니다.

 

n = 1 : num_list의 0번 인덱스부터 b번 인덱스까지

n = 2 : num_list의 a번 인덱스부터 마지막 인덱스까지

n = 3 : num_list의 a번 인덱스부터 b번 인덱스까지

n = 4 : num_list의 a번 인덱스부터 b번 인덱스까지 c 간격으로

올바르게 슬라이싱한 리스트를 return하도록 solution 함수를 완성해주세요.

---

제한사항

 

n 은 1, 2, 3, 4 중 하나입니다.

slicer의 길이 = 3

slicer에 담긴 정수를 차례대로 a, b, c라고 할 때

• 0 ≤ a ≤ b ≤ num_list의 길이 - 1
• 1 ≤ c ≤ 3

5 ≤ num_list의 길이 ≤ 30

0 ≤ num_list의 원소 ≤ 100

---

입출력 예

 

n	slicer	num_list	result
3	[1,5,2]	[1,2,3,4,5,6,7,8,9]	[2,3,4,5,6]
4	[1,5,2]	[1,2,3,4,5,6,7,8,9]	[2,4,6]

---

입출력 예 설명

 

입출력 예 #1

[1, 2, 3, 4, 5, 6, 7, 8, 9]에서 1번 인덱스부터 5번 인덱스까지 자른 리스트는 [2, 3, 4, 5, 6]입니다.

 

입출력 예 #2

[1, 2, 3, 4, 5, 6, 7, 8, 9]에서 1번 인덱스부터 5번 인덱스까지 2개 간격으로 자른 리스트는 [2, 4, 6]입니다.

---

풀이

 

-> 조건문으로 n 값이 1,2,3,4 일 때로 나누고, n=4일 경우 slicer[1]-slicer[0]값이 홀수일 때와 짝수일 때를 나누어 계산한다.

---

class Solution {
    public int[] solution(int n, int[] slicer, int[] num_list) {
        int[] answer = {};
        if(n==1){
            answer = new int[slicer[1]+1];
            for(int i=0; i<slicer[1]+1;i++){
                answer[i]=num_list[i];
            }
        } else if(n==2){
            answer = new int[num_list.length-slicer[0]];
            for(int i=0; i<num_list.length-slicer[0]; i++){
                answer[i]=num_list[slicer[0]+i];
            }
        } else if(n==3){
            answer = new int[slicer[1]-slicer[0]+1];
            for(int i=0; i<slicer[1]-slicer[0]+1;i++){
                answer[i]=num_list[slicer[0]+i];
            }
        } else if(n==4){
            if((slicer[1]-slicer[0])%2==0){
                answer = new int[(slicer[1]-slicer[0])/2+1];
                for(int i=0; i<answer.length;i++){
                    answer[i]=num_list[slicer[0]+2*i];
                }
            } else if((slicer[1]-slicer[0])%2==1){
                answer = new int[(slicer[1]-slicer[0])/2+1];
                for(int i=0; i<answer.length; i++){
                    answer[i]=num_list[slicer[0]+2*i];
                }
            }
        }
        return answer;
    }
}