[프로그래머스/C언어] 2의 영역

문제

정수 배열 arr가 주어집니다. 배열 안의 2가 모두 포함된 가장 작은 연속된 부분 배열을 return 하는 solution 함수를 완성해 주세요.

단, arr에 2가 없는 경우 [-1]을 return 합니다.

 

솔루션

#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(arr_len * sizeof(int));
    int start, end;
    start = -1;
    int i, j;
    for(int i = 0; i < arr_len; i++)
        if(arr[i] == 2){
            start = i;
            break;
        }
    for(int j = arr_len - 1; j >= 0; j--)
        if(arr[j] == 2){
            end = j;
            break;
        }
    printf("%d %d", start, end);
    for(i = start; i <= end; i++)
        answer[i - start] = arr[i];
    
    if(start == -1)
        *answer = -1;
    return answer;
}

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

프로그래머스