프로그래머스 - 잘라서 배열로 저장하기 / JAVA (자바) 코딩 알고리즘 연습

반응형

class Solution {
    public String[] solution(String my_str, int n) {
		int length = 0;
		
		if(my_str.length() % n > 0) {
			length = (my_str.length() / n) + 1;
		}else if(my_str.length() % n == 0) {
			length = my_str.length() / n;
		}
		
		int a = my_str.length() / n; //길이 나눈값
		int b = my_str.length() % n; //길이 나머지
		
		String[] answer = new String[length];
		
		int idx1 = 0;
		int idx2 = n;
		
		for (int i = 0; i <= a; i++) {
			if(i==a & my_str.length() % n > 0) {
				answer[a] = my_str.substring(idx1,my_str.length());
				break;
			}else if(i==a) {
				break;
			}
			answer[i] = my_str.substring(idx1,idx2);
			idx1 = idx1+n;
			idx2 += n;
		}
        
        return answer;
    }
}