프로그래머스 잘라서 배열로 저장하기[Java]

잘라서 배열로 저장하기

💫문제 설명💫

문자열 my_str과 n이 매개변수로 주어질 때,
my_str을 길이 n씩 잘라서 저장한 배열을 return하도록 solution 함수를 완성해주세요.

💫제한사항💫

1 ≤ my_str의 길이 ≤ 100
1 ≤ n ≤ my_str의 길이
my_str은 알파벳 소문자, 대문자, 숫자로 이루어져 있습니다.

💫입출력 ex💫

my_str	n	result
“abc1Addfggg4556b”	6	[“abc1Ad”, “dfggg4”, “556b”]
“abcdef123”	3	[“abc”, “def”, “123”]

문제풀이

class Solution {
    public String[] solution(String my_str, int n) {
        String[] answer = {};
        
        if(my_str.length() % n == 0) {
            answer = new String[my_str.length()/n];
        } else {
            answer = new String[my_str.length()/n+1];
        }
        
        // substring 문자열 자르는 메서드 (start, end)
        for(int i=0; i<answer.length; i++) {
            answer[i] = (
                i*n+n <= my_str.length() - 1 ? 
                my_str.substring(n*i,n*i+n) : my_str.substring(n*i)
            );
        }
        return answer;
    }
}