[프로그래머스(Java)] 왼쪽 오른쪽 / Stream

 

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

 

풀이

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

 

해결방법 

1. 리스트를탐색하여 "r" 또는 "l"이 최초로 등장한 index를 구하고, 해당 문자를 저장한다.

2. 문자가 "r"일 경우 (index, r]까지, "l"일 경우 [0, index)까지의 배열을 구한다. 

 

 

---

다른 풀이

import java.util.Arrays;
import java.util.stream.IntStream;

class Solution {
    public String[] solution(String[] str_list) {
        return IntStream.range(0, str_list.length)
                .boxed()
                .filter(i -> str_list[i].equals("l") || str_list[i].equals("r"))
                .findFirst()
                .map(i -> {
                    if (str_list[i].equals("l")) {
                        return Arrays.copyOfRange(str_list, 0, i);
                    }
                    return Arrays.copyOfRange(str_list, i + 1, str_list.length);
                })
                .orElseGet(() -> new String[]{});
    }
}

 

boxed() 

IntStream.range(0, 3).boxed() // => Stream<Integer> 생성

• int, long, double 요소를 Integer, Long, Double 요소로 박싱하여 스트림을 생성한다. 

 

findFirst()

• filter 조건에 일치하는 element 1개를 리턴한다.
• 조건에 일치하는 요소가 없다면 empty가 리턴된다. 

 

 

---

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

프로그래머스