프로그래머스 181912번 배열 만들기 5 - Python, Java
문제 설명
문자열 배열 intStrs와 정수 k, s, l가 주어집니다. intStrs의 원소는 숫자로 이루어져 있습니다.
배열 intStrs의 각 원소마다 s번 인덱스에서 시작하는 길이 l짜리 부분 문자열을 잘라내 정수로 변환합니다. 이때 변환한 정수값이 k보다 큰 값들을 담은 배열을 return 하는 solution 함수를 완성해 주세요.
제한 사항
- 0 ≤
s< 100 - 1 ≤
l≤ 8 - 10^(l - 1) ≤
k< 10^l - 1 ≤
intStrs의 길이 ≤ 10,000s+l≤intStrs의 원소의 길이 ≤ 120
입출력 예시
| intStrs | k | s | l | result |
|---|---|---|---|---|
| [“0123456789”,”9876543210”,”9999999999999”] | 50000 | 5 | 5 | [56789, 99999] |
입출력 예 설명 1
idx에 따라 잘라낸 문자열과 그에 따른 ret의 변화를 표시하면 다음 표와 같습니다.
| idx | 잘라낸 문자열 | idx |
|---|---|---|
| 0 | “56789” | [56789] |
| 1 | “43210” | [56789] |
| 2 | “99999” | [56789, 99999] |
코드 구현
Python
def solution(intStrs, k, s, l):
answer = []
for intStr in intStrs:
intStr = int(intStr[s:s+l])
if intStr > k:
answer.append(intStr)
return answer
Java
import java.util.ArrayList;
class Solution {
public int[] solution(String[] intStrs, int k, int s, int l) {
ArrayList<Integer> answer = new ArrayList<>();
for (String intStr : intStrs) {
int slicedIntStr = Integer.parseInt(intStr.substring(s, s + l));
if (slicedIntStr > k) {
answer.add(slicedIntStr);
}
}
return answer.stream().mapToInt(i -> i).toArray();
}
}