[프로그래머스 코딩기초 트레이닝] 왼쪽 오른쪽 (JAVA)

문제 설명

 

문자열 리스트 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합니다.

---

풀이

 

-> 먼저 str_list에서 l과 r 값이 몇번째 인덱스에서 처음 나오는지 count와 pointer 변수를 통해서 구해주었다. 그 다음 pointer가 l일 때와 r일 때를 나누어서 answer를 선언해주고, 반복문을 통해서 각 경우의 배열 크기만큼 str_list에서 answer배열에 값을 담았다. 이 때, pointer가 l 일 때는, count가 0이 아니고, pointer가 r 일 때는, count가 str_list의 길이와 같으면 안된다는 것을 표현해주었다.

---

코드

class Solution {
    public String[] solution(String[] str_list) {
        String[] answer = {};
        int count = 0;
        String pointer = "";
        for(int i=0;i<str_list.length;i++){
            if(str_list[i].equals("l")||str_list[i].equals("r")){
                pointer = str_list[i];
                break;
            }
            count++;
        }
        
        if(pointer.equals("l")){
            if(count!=0){
                answer = new String[count];
                for(int i=0;i<count;i++){
                    answer[i]=str_list[i];
                }
            }
        } else if(pointer.equals("r")) {
            if(count!=str_list.length-1){
                answer = new String[str_list.length-count-1];
                for(int i=0;i<str_list.length-count-1;i++){
                    answer[i]=str_list[count+1+i];
                }
            }
        }
        
        return answer;
    }
}