[프로그래머스0] 배열 자바

문제 설명

문자열 배열 intStrs와 정수 k, s, l가 주어집니다. intStrs의 원소는 숫자로 이루어져 있습니다.

배열 intStrs의 각 원소마다 s번 인덱스에서 시작하는 길이 l짜리 부분 문자열을 잘라내 정수로 변환합니다. 이때 변환한 정수값이 k보다 큰 값들을 담은 배열을 return 하는 solution 함수를 완성해 주세요.

---

제한사항

• 0 ≤ s < 100
• 1 ≤ l ≤ 8
• 10 ≤ k < 10l
• l - 1
• 1 ≤ intStrs의 길이 ≤ 10,000
  • s + l ≤ intStrs의 원소의 길이 ≤ 120

---

입출력 예

intStrs k s l result

["0123456789","9876543210","9999999999

 

 

import java.util.ArrayList;

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

        for (int i = 0; i < intStrs.length; i++) {
            String substring = intStrs[i].substring(s, s + l);

            if (Integer.parseInt(substring) > k) {
                resultList.add(Integer.parseInt(substring));
            }
        }

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

        return answer;
    }
}

 

유효한 값이 있는 경우에만 배열에 담게하였다. 

처음에는 다음과 같이 넣으려 했었다.

 

class Solution {
    public int[] solution(String[] intStrs, int k, int s, int l) {
       int[] answer = new int[intStrs.length]; 
        
        for (int i = 0; i < intStrs.length; i++) {
            String substring = intStrs[i].substring(s, s + l);
            answer[i] = Integer.parseInt(substring); 
        }
        return answer;
}
}

그랬더니 {~~,0.~~} 이렇게 0값이 담겼다. 

 

 

python으로 풀었을 때에는 다음과 같이 풀 수 있다. 

def solution(intStrs, k, s, l):
    return [int(intstr[s:s+l]) for intstr in intStrs if int(intstr[s:s+l]) > k]