자바 알고리즘 - 잘라서 배열로 저장하기

문제 설명

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

 

제한 사항

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

 

입출력 예

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

 

답안

class Solution {
    public String[] solution(String my_str, int n) {
        int idx = my_str.length()%n==0?my_str.length()/n:my_str.length()/n+1;
        String[] answer = new String[idx];
        

        for(int j=0; j<idx; j++)
        {
            int first = n*j;
            int last = (first+n)>=my_str.length()?my_str.length():first+n;
            
            answer[j]=my_str.substring(first,last);
        }
        
        return answer;
    }
}

먼저 n개만큼 my_str을 끊었을 때 배열이 몇 개가 만들어지는지 계산한다.

그리고 answer 배열을 그 갯수에 맞게 초기화 한뒤 배열 갯수만큼 for문을 돌면서 substring을 통해 끊어줄 문자열 인덱스 값을 계산한다.