[코딩테스트 Lv.0] 2의 영역

문제 설명

정수 배열 arr가 주어집니다. 배열 안의 2가 모두 포함된 가장 작은 연속된 부분 배열을 return 하는 solution 함수를 완성해 주세요.단, arr에 2가 없는 경우 [-1]을 return 합니다.

제한사항

● 1 ≤ arr의 길이 ≤ 100,000
   ○ 1 ≤ arr의 원소 ≤ 10

입출력 예

arr				result
[1, 2, 1, 4, 5, 2, 9]		[2, 1, 4, 5, 2]
[1, 2, 1]			[2]
[1, 1, 1]			[-1]
[1, 2, 1, 2, 1, 10, 2, 1]	[2, 1, 2, 1, 10, 2]

입출력 설명

입출력 예 #1
2가 있는 인덱스는 1번, 5번 인덱스뿐이므로 1번부터 5번 인덱스까지의 부분 배열인 [2, 1, 4, 5, 2]를 return 합니다.

입출력 예 #2
2가 한 개뿐이므로 [2]를 return 합니다.

입출력 예 #3
2가 배열에 없으므로 [-1]을 return 합니다.

입출력 예 #4
2가 있는 인덱스는 1번, 3번, 6번 인덱스이므로 1번부터 6번 인덱스까지의 부분 배열인 [2, 1, 2, 1, 10, 2]를 return 합니다.

---

내 풀이

Java

class Solution {
    public int[] solution(int[] arr) {
        int[] answer = {};
        
        int s = -1, e = -1;
        
        for(int i = 0; i < arr.length; i++){
            if(s == -1 && arr[i] == 2){
                s = i;
                e = i;
            }else if(arr[i] == 2){
                e = i;
            }
        }
        
        if(s == -1 && e == -1) return new int[]{-1};
        answer = new int[e - s + 1];
        for(int i = s, k = 0; i <= e; i++){
            answer[k++] = arr[i];
        }
        
        return answer;
    }
}

C

#include <stdio.h>
#include <stdbool.h>
#include <stdlib.h>

// arr_len은 배열 arr의 길이입니다.
int* solution(int arr[], size_t arr_len) {
    // return 값은 malloc 등 동적 할당을 사용해주세요. 할당 길이는 상황에 맞게 변경해주세요.
    int* answer = (int*)malloc(sizeof(int) * arr_len);
    int s = -1, e = -1;
    int i;
    for(i = 0; i < arr_len; i++){
        if(s == -1 && arr[i] == 2){
            s = i;
            e = i;
        }else if(arr[i] == 2){
            e = i;
        }
    }
    
    if(s == -1 && e == -1) {
        answer[0] = -1;
        return answer;
    }
    
    int k;
    for(i = s, k = 0; i <= e; i++){
        answer[k++] = arr[i];
    }

    return answer;
}

Python

def solution(arr):
    answer = []
    
    if 2 not in arr:
        return [-1]
    
    answer = arr[arr.index(2):len(arr) - arr[::-1].index(2)]
    
    return answer

---

https://school.programmers.co.kr/learn/courses/30/lessons/181894

프로그래머스