본문 바로가기

코딩테스트 스터디

코딩테스트 연습 - 배열 원소의 길이 (JAVA)

프로그래머스 코딩테스트 입문 0단계 문제입니다

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

📢 문제 설명

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

 

⭕ 정답 코드

1. Stream 사용

class Solution {
    public int[] solution(String[] strlist) {
        int[] answer = new int[strlist.length];

        for(int i = 0; i<strlist.length; i++)
            answer[i] = strlist[i].length();

        return answer;

    }
}

직접 푼 풀이입니다

2. 반복문으로 풀기

class Solution {
    public int[] solution(String[] strlist) {
    
        return Arrays.stream(strlist) //문자열 배열로 Stream 생성
                .mapToInt(String::length) //int형으로 형변환, String 클래스의 length() 메서드
                .toArray(); //배열로 바꾸기
    }
}

설명은 주석을 참고해주세요

Stream 힌트를 얻어서 혼자 풀어본 풀이입니다

 

✅ 회고

Stream을 쓸 수 있는 문제인지 조금 더 생각해보자!