[코딩테스트 Lv.0] 왼쪽 오른쪽

문제 설명

문자열 리스트 str_list에는 "u", "d", "l", "r" 네 개의 문자열이 여러 개 저장되어 있습니다. str_list에서 "l"과 "r" 중 먼저 나오는 문자열이 "l"이라면 해당 문자열을 기준으로 왼쪽에 있는 문자열들을 순서대로 담은 리스트를, 먼저 나오는 문자열이 "r"이라면 해당 문자열을 기준으로 오른쪽에 있는 문자열들을 순서대로 담은 리스트를 return하도록 solution 함수를 완성해주세요. "l"이나 "r"이 없다면 빈 리스트를 return합니다.

제한사항

1 ≤ str_list의 길이 ≤ 20
str_list는 "u", "d", "l", "r" 네 개의 문자열로 이루어져 있습니다.

입출력 예

str_list		result
["u", "u", "l", "r"]	["u", "u"]
["l"]			[]

입출력 설명

입출력 예 #1
"r"보다 "l"이 먼저 나왔기 때문에 "l"의 왼쪽에 있는 문자열들을 담은 리스트인 ["u", "u"]를 return합니다.

입출력 예 #2
"l"의 왼쪽에 문자열이 없기 때문에 빈 리스트를 return합니다.

---

내 풀이

Java

class Solution {
    public String[] solution(String[] str_list) {
        String[] answer = {};
        
        int s = 0, e = 0;
        
        for(int i = 0; i < str_list.length; i++){
            if(str_list[i].equals("l")){
                e = i;                
                break;
            }
            if(str_list[i].equals("r")){
                s = i + 1;
                e = str_list.length;
                break;
            }
        }
        
        if(s == 0 && e == 0) return answer;
        
        answer = new String[e - s];
        
        int k = 0;
        for(int i = s; i < e; i++){
            answer[k++] = str_list[i];
        }
        
        return answer;
    }
}

C

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

// str_list_len은 배열 str_list의 길이입니다.
// 파라미터로 주어지는 문자열은 const로 주어집니다. 변경하려면 문자열을 복사해서 사용하세요.
char** solution(const char* str_list[], size_t str_list_len) {
    // return 값은 malloc 등 동적 할당을 사용해주세요. 할당 길이는 상황에 맞게 변경해주세요.
    
    int i;    
    int s = 0, e = str_list_len;
    
    for(i = 0; i < str_list_len; i++){
        if(str_list[i][0] == 'l'){
            e = i;
            break;
        } 
        if(str_list[i][0] == 'r'){
            s = i + 1;
            break;
        }
    }
    char** answer = (char**)malloc(sizeof(char*) * (e - s));
    int k;
    for(i = s, k = 0; i < e; i++){
        answer[k++] = str_list[i];
    }
    
    return answer;
}

Python

def solution(str_list):
    answer = []
    
    for i in range(len(str_list)):
        if str_list[i] == 'l':
            return str_list[:i]
        elif str_list[i] == 'r':
            return str_list[i+1:]
    
    return answer

---

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

프로그래머스