Programmers | 코딩테스트 입문 - 배열 자르기, 파이썬 slice 기능

프로그래머스 Lv.0 배열 자르기

 

문제 설명

정수 배열 numbers와 정수 num1, num2가 매개변수로 주어질 때, numbers의 num1번 째 인덱스부터 num2번째 인덱스까지 자른 정수 배열을 return 하도록 solution 함수를 완성해보세요.

---

제한사항

• 2 ≤ numbers의 길이 ≤ 30
• 0 ≤ numbers의 원소 ≤ 1,000
• 0 ≤num1 < num2 < numbers의 길이

 

JAVA

class Solution {
    public int[] solution(int[] numbers, int num1, int num2) {
        int idx = 0;
        int[] answer = new int[num2-num1+1];
        
        for(int i=num1; i<=num2; i++){
            answer[idx++] = numbers[i];
        }
        return answer;
    }
}

 

Python3

def solution(numbers, num1, num2):
    answer = []
    for i in range(num1,num2+1):
        answer.append(numbers[i])
    return answer

 

다른 사람 풀이

def solution(numbers, num1, num2):
    return numbers[num1:num2+1]

• 파이썬의 slice기능

• slice(시작,끝) : 시작인덱스부터 (끝-1) 인덱스까지 자른다.
• 배열에서는 [시작:끝] 형식으로 사용한다.(동일하게 끝 미포함)
• *cf) 배열[시작:끝:증가폭] : 시작인덱스부터 (끝-1)인덱스까지 증가폭만큼 인덱스가 증가.
• ex) numbers[0,1,2,3,4,5,6,7,8,9,10]이 있고, numbers[0:10:3] = [0,3,6,9]를 출력