잘라서 배열로 저장하기

날짜 : 2024년 6월 24일

기술 : array

언어 : C++

출처 : https://school.programmers.co.kr/learn/courses/30/lessons/120913 programmers

문제 설명

문자열 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해야 합니다.

유의사항

  • 입출력 예 #1의 경우 “abc1Addfggg4556b”를 길이 6씩 자르면 “abc1Ad”, “dfggg4” 두개와 마지막 “556b”가 남습니다. 이런 경우 남은 문자열을 그대로 배열에 저장합니다.

Solution

#include <iostream>
#include <string>
#include <vector>

using namespace std;

vector<string> solution(string my_str, int n) {
    vector<string> answer;

    // 0부터 문자열의 길이까지 n씩 증가시키며 반복
    for (int i = 0; i < my_str.length(); i += n) {
        // i부터 n개의 문자를 잘라서 vector에 추가
        answer.push_back(my_str.substr(i, n));
    }

    return answer;
}

int print(vector<string> result) {

    for (string str : result) {
        cout << str << " ";
    }
    cout << endl;

    return 0;
}

int main() {
    print(solution("abc1Addfggg4556b", 6));
    print(solution("abcdef123", 3));
}

목차