[C++] 프로그래머스 세로 읽기

문제설명

문자열 my_string과 두 정수 m, c가 주어집니다.
my_string을 한 줄에 m 글자씩 가로로 적었을 때
왼쪽부터 세로로 c번째 열에 적힌 글자들을
문자열로 return 하는 solution 함수를 작성해 주세요.

제한사항

• my_string은 영소문자로 이루어져 있습니다.
• 1 ≤ m ≤ my_string의 길이 ≤ 1,000
• m은 my_string 길이의 약수로만 주어집니다.
• 1 ≤ c ≤ m

입출력 예

my_string	m	c	result
“ihrhbakrfpndopljhygc”	4	2	“happy”
“programmers”	1	1	“programmers”

풀이

my_string 문자열을 m의 수만큼 가로로 나열하는 배열을 만든 뒤 c열의 글자들을 리턴하는 로직을 구현하여 문제를 풀 수 있었습니다.

#include <string>
#include <vector>

using namespace std;

string solution(string my_string, int m, int c) {
    string answer = "";
    vector<vector<char>> arr;
    int idx = 0;
    
    for(int i = 0; i < my_string.length()/m; i++)
    {
        vector<char> tmp;
        for(int j = 0; j < m; j++)
        {
            tmp.push_back(my_string[idx]);
            idx++;
        }
        
        arr.push_back(tmp);
        
        answer.push_back(arr[i][c - 1]);
    }
    
    return answer;
}

결과