[프로그래머스] Lv.0 배열 만들기 5.java

배열 만들기 5.java

import java.util.ArrayList;
import java.util.List;

class Solution {
    public int[] solution(String[] intStrs, int k, int s, int l) {
        List<Integer> result = new ArrayList<>();

        for (String str : intStrs) {
            if (str.length() >= s + l) {
                String substring = str.substring(s, s + l);
                int num = Integer.parseInt(substring);
                if (num > k) {
                    result.add(num);
                }
            }
        }

        int[] answer = new int[result.size()];
        for (int i = 0; i < result.size(); i++) {
            answer[i] = result.get(i);
        }

        return answer;
    }
}

result ArrayList는 조건을 만족하는 정수들을 저장하기 위한 용도
intStrs 배열의 각 원소의 길이가 s+1보다 크거나 같은지 확인 후 k보다 크면 result 리스트에 추가
result 리스트에 저장된 정수들을 배열로 변환하여 answer 반환

출처: 프로그래머스 코딩 테스트 연습, https://school.programmers.co.kr/learn/challenges