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

문제

링크

---

코드

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]);
    }
}

---

리뷰

주어진 문자열을 n단위로 끊어서 배열로 저장하는 문제이다.
문자열의 길이가n값에 딱 나누어 떨어지는 경우 반복문만 통과하면 완성되겠지만
딱 나누어 떨어지지 않는 경우 마지막 잘린문자는 배열에 포함되지 않는다.

if문을 통해 딱나누어 떨어지지 않는 경우에 뒷부분의 남은 문자열이 입력되도록 하였다.