Your Name

I am an amazing person.

2 분 소요

문제 설명

프로그래머스 181897번 리스트 자르기

정수 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]입니다.

코드 구현

Python

def solution(n, slicer, num_list):
    answer = []

    if n == 1:
        answer = num_list[:slicer[1] + 1]
    elif n == 2:
        answer = num_list[slicer[0]:]
    elif n == 3:
        answer = num_list[slicer[0]:slicer[1] + 1]
    elif n == 4:
        temp = []
        for i in range(slicer[0], slicer[1] + 1, slicer[2]):
            temp.append(num_list[i])
        answer = temp

    return answer

Java

import java.util.ArrayList;

class Solution {
    public int[] solution(int[] arr) {
        int first2 = -1;
        int last2 = -1;

        ArrayList<Integer> answer = new ArrayList<>();

        for (int index = 0; index < arr.length; index += 1) {
            if (first2 == -1 && arr[index] == 2) {
                first2 = index;
            } else if (first2 != -1 && arr[index] == 2) {
                last2 = index;
            }
        }

        if (first2 != -1 && last2 == -1) {
            answer.add(arr[first2]);
        } else if (first2 != -1 && last2 != -1) {
            for (int i = first2; i <= last2; i += 1) {
                answer.add(arr[i]);
            }
        } else {
            answer.add(-1);
        }

        return answer.stream().mapToInt(i -> i).toArray();
    }
}

개선할 점 - Python

  • 배열을 자를 때, [a:b:c]로 하면 c 간격으로 자를 수 있다.
def solution(n, slicer, num_list):
    answer = []

    if n == 1:
        answer = num_list[:slicer[1] + 1]
    elif n == 2:
        answer = num_list[slicer[0]:]
    elif n == 3:
        answer = num_list[slicer[0]:slicer[1] + 1]
    elif n == 4:
        answer = num_list[slicer[0]:slicer[1] + 1:slicer[2]]

    return answer

참고