[Programmers] 잘라서 배열로 저장하기 자바

문제설명

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

 

 

입출력 예

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

 

 

통과한 코드 ✅

import java.util.ArrayList;
import java.util.List;
class Solution {
    public String[] solution(String my_str, int n) {
     List<String> answer=new ArrayList();
        for(int i=0; i<(my_str.length()/n); i++){
            answer.add(my_str.substring((n*i),n*(i+1)));
        }
        if(my_str.length()%n!=0){
            answer.add(my_str.substring((my_str.length()/n)*n));
        }
        return answer.toArray(new String[0]);
    }
}

my_str의 길이 만큼 딱 떨어지지 않는 경우를  생각해서 if문을 작성했다. 

 

 

 

입출력 결과