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

잘라서 배열로 저장하기

문제 설명

문자열 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”]

---

입출력 예 설명

입출력 예 #1

• “abc1Addfggg4556b” 를 길이 6씩 잘라 배열에 저장한 [“abc1Ad”, “dfggg4”, “556b”]를 return해야 합니다.

입출력 예 #2

• “abcdef123” 를 길이 3씩 잘라 배열에 저장한 [“abc”, “def”, “123”]를 return해야 합니다.

---

나의 풀이 코드

import java.util.*;
class Solution {
    public String[] solution(String my_str, int n) {
        String[] answer = {};
        int cnt = 0;
        String temp_str = "";
        String[] str_arr = new String[(int)Math.ceil((double)my_str.length()/n)];
        
        for (int i=0; i<my_str.length(); i++){
            temp_str = temp_str.concat(String.valueOf(my_str.charAt(i)));
            cnt++;
            if (cnt == n || i == my_str.length() - 1) {
                str_arr[i / n] = temp_str; 
                temp_str = "";
                cnt = 0;
            }
        }
        return str_arr;
    }
}

각 반복에서 요소들을 temp문자열에 넣고 cnt를 증가시킨다.
cnt가 n이거나 문자열이 끝났다는 조건을 만족하면 temp문자열을 배열에 삽입하고 cnt를 초기화한다.
반복한다.

배운점

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()으로 문자열을 부분적으로 추출하기 위해서 인덱스(start,end)를 구하고 추출하는것을 반복한다.