[프로그래머스]배열 원소의 길이

배열 원소의 길이

문제 설명

문자열 배열 strlist가 매개변수로 주어집니다. strlist 각 원소의 길이를 담은 배열을 retrun하도록 solution 함수를 완성해주세요.

---

제한사항

• 1 ≤ strlist 원소의 길이 ≤ 100
• strlist는 알파벳 소문자, 대문자, 특수문자로 구성되어 있습니다.

---

입출력 예

strlist	result
[“We”, “are”, “the”, “world!”]	[2, 3, 3, 6]
[“I”, “Love”, “Programmers.”]	[1, 4, 12]

---

입출력 예 설명

입출력 예 #1

• [“We”, “are”, “the”, “world!”]의 각 원소의 길이인 [2, 3, 3, 6]을 return합니다.

입출력 예 #2

• [“I”, “Love”, “Programmers.”]의 각 원소의 길이인 [1, 4, 12]을 return합니다.

---

나의 풀이 코드

import java.util.*;

class Solution {
    public int[] solution(String[] strlist) {
        int[] answer = {};
        
        List<Integer> size_list = new ArrayList<>();
        
        for(int i=0; i<strlist.length; i++){
            size_list.add(strlist[i].length());
        }   
        
        answer = size_list.stream().mapToInt(Integer::intValue).toArray();
        
        return answer;
    }
}

배열을 사용하지 않고 리스트를 사용해서 풀어보기로 했다.

size_list에 각 요소의 크기를 넣고 이를 스트림을 이용하여 answer배열로 바꾸었다.

배운점

import java.util.Arrays;

class Solution {
    public int[] solution(String[] strList) {
        return Arrays.stream(strList).mapToInt(String::length).toArray();
    }
}

다른 사람이 푼 코드인데 내 마지막 코드에서 intValue를 length로만 바꾸어 간단하게 풀었다.
스트림에 더욱 익숙해져야겠다.