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

💡 문제 

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

 

 

 

문제 요구사항:

• 문자열로 이루어진 리스트가 주어진다.
• 각 문자열의 길이를 구하여, 새로운 리스트에 담아 반환한다.

해결 방법:

• 각 문자열의 길이를 구하는 방법은 len() 함수를 사용하면 된다.
• 문자열로 이루어진 리스트를 반복문으로 순회하면서 각 원소의 길이를 구하고, 그 값을 새로운 리스트에 담아서 반환하면 된다.

 

 

🖍 내 답안

public int[] solution(String[] strlist) { // String 배열을 입력으로 받고, int 배열을 반환
        // 입력값으로 null이나 빈 배열이 들어올 때, solution 메서드가 예외를 던지지 않고 빈 배열을 반환하는지 검증
        if (strlist == null) {
            return new int[0];
        }

        int[] lengths = new int[strlist.length]; // 길이가 strlist.length인 int 배열 생성
        for (int i = 0; i < strlist.length; i++) { // strlist 배열의 각 원소에 대해 length() 메서드를 호출하여 길이를 구한다.
            lengths[i] = strlist[i].length(); // lengths 배열에 저장하고, 마지막으로 lengths 배열을 반환
        }
        return lengths;
    }

public int[] solution2(String[] strings) {
        return Arrays.stream(strings)// strings 배열을 stream으로 변환한 뒤
                    .mapToInt(String::length) // mapToInt 메서드를 이용해 String 요소를 int로 매핑
                    .toArray(); // 최종적으로 toArray 메서드를 호출하여 int[] 배열로 변환
    }