Algorithm/프로그래머스

[프로그래머스] 리스트 자르기

Llimy1 2024. 3. 9. 01:27
반응형
SMALL
반응형
SMALL

[level 0] 리스트 자르기 - 181897

문제 링크

성능 요약

메모리: 73.2 MB, 시간: 0.02 ms

구분

코딩테스트 연습 > 코딩 기초 트레이닝

채점결과

정확성: 100.0
합계: 100.0 / 100.0

제출 일자

2024년 03월 09일 01:25:48

문제 설명

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

  • n = 1 : num_list의 0번 인덱스부터 b번 인덱스까지
  • n = 2 : num_lista번 인덱스부터 마지막 인덱스까지
  • n = 3 : num_lista번 인덱스부터 b번 인덱스까지
  • n = 4 : num_lista번 인덱스부터 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]입니다.

출처: 프로그래머스 코딩 테스트 연습, https://school.programmers.co.kr/learn/challenges

 

문제 정답

class Solution {
    public int[] solution(int n, int[] slicer, int[] num_list) {
        int size = 0;
        int[] answer = new int[0];
        switch (n) {
            case 1 -> {
                int b = slicer[1];
                size = b + 1;
                answer = new int[size];
                for (int i = 0; i <= b; i++) {
                    answer[i] = num_list[i];
                }
            }
            case 2 -> {
                int a = slicer[0];
                int len = num_list.length;
                size = len - a ;
                answer = new int[size];
                for (int i = a; i < len; i++) {
                    answer[i - a] = num_list[i];
                }
            }
            case 3 -> {
                int a = slicer[0];
                int b = slicer[1];
                size = b - a + 1;
                answer = new int[size];
                for (int i = a; i <= b; i++) {
                    answer[i - a] = num_list[i];
                }
            }
            case 4 -> {
                int a = slicer[0];
                int b = slicer[1];
                int c = slicer[2];
                int count = 0;
                size = (b - a) / 2 + 1;
                answer = new int[size];
                for (int i = a; i <= b; i += c) {
                    answer[count++] = num_list[i];
                }
            }
        }
        return answer;
    }
}
반응형
LIST