잘라서 배열로 저장하기

문제 설명

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

제한사항

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

나의 코드

import java.util.List;
import java.util.ArrayList;

class Solution {
    public String[] solution(String my_str, int n) {
        List<String> list = new ArrayList<>();
        for(int i=0; i<my_str.length(); i+=n) {
            if(i+n > my_str.length()) {
                list.add(my_str.substring(i));
            } else {
                list.add(my_str.substring(i, i+n));
            }
        }
        String[] answer = new String[list.size()];
        for(int i=0; i<list.size(); i++) {
            answer[i] = list.get(i);
        }
        return answer;
    }
}

다른 사람 코드

class Solution {
    public String[] solution(String my_str, int n) {
        int resultCnt = (my_str.length() + n - 1) / n;
        String[] answer = new String[resultCnt];

        for (int i = 0; i < resultCnt; i++) {
            int start = n * i;
            int end = start + n >= my_str.length()? my_str.length(): start + n;
            answer[i] = my_str.substring(start, end);
        }

        return answer;
    }
}

느낀 점

반복문을 돌면서 substring()메서드를 활용하여 풀었다. 처음에는 i+n단위로 커지다 보니 마지막에 남은 문자열을 처리할 수 없어 실패했는데, 조건문을 사용하여 마지막 남은 문자열까지 list에 추가해주면서 해결했다.
List를 사용하여 풀었지만, 다른 사람 코드를 보니 충분히 문자열 배열을 가지고도 풀 수 있었다.