[프로그래머스] 배열 만들기 5 (Java, Python)

문제

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

프로그래머스

 

코드(Java)

import java.util.*;

class Solution {
    public int[] solution(String[] intStrs, int k, int s, int l) {
        List<Integer> answer = new ArrayList<>();
        
        for (String str : intStrs) {
            int i = Integer.parseInt(str.substring(s, s + l));
            if (i > k) {
                answer.add(i);
            }
        }
        
        return answer.stream().mapToInt(i->i).toArray();
    }
}

 

코드(Python)

def solution(intStrs, k, s, l):
    answer = []
    for string in intStrs:
        num = int(string[s:s+l])
        if num > k:
            answer.append(num)
    return answer