자바 왼쪽 오른쪽 (코딩테스트-연습코딩-기초 트레이닝-왼쪽 오른쪽)

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

프로그래머스

---

문제 설명

문자열 리스트 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" 네 개의 문자열로 이루어져 있습니다.

---

입출력 예

["u", "u", "l", "r"]	["u", "u"]
["l"]	[]

---

입출력 예 설명

입출력 예 #1

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

입출력 예 #2

• "l"의 왼쪽에 문자열이 없기 때문에 빈 리스트를 return합니다.

---

풀이코드 

import java.util.*;

class Solution {
    public String[] solution(String[] str_list) {
        List<String> list = new ArrayList<>();

        for (int i = 0; i < str_list.length; i++) { //처음 배열 저장
            if (!str_list[i].equals("l") && !str_list[i].equals("r")) {
                list.add(str_list[i]);
                
            } else if (str_list[i].equals("l")) { //l이므로 저장된 배열 반환
                return list.toArray(new String[0]);
                
            } else if (str_list[i].equals("r")) { 
                // r이면 배열 삭제 후 오른쪽 배열 다시 저장
                list.clear(); // Clear the list when "r" is encountered
                for (int j = i + 1; j < str_list.length; j++) {
                    list.add(str_list[j]);  
                }
                // 저장된 배열 반환 
                return list.toArray(new String[0]);
            }
        }

        return new String[0];
    }
}

 

풀이방법 

배열에서 for문을 실행하여 l과 r이 나오지 않으면 조건문을 통해서 배열을 리스트에 저장해준다.
l이 나온다면 저장된 리스트를 반환해준다.

만약 r이 나온다면 저장된 리스트를 삭제해주고,
다시 실행하던 i번째부터 for문을 실행하여 r의 오른쪽 배열을 리스트에 저장한다.
그리고 저장된 리스트를 반환해준다.

l과 r이 나오지 않는 경우에는 빈 배열을 반환해준다.