[코딩테스트 Lv.0] 세로 읽기

문제 설명

문자열 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"

입출력 설명

입출력 예 #1
예제 1번의 my_string을 한 줄에 4 글자씩 쓰면 다음의 표와 같습니다.
1열	2열	3열	4열
i	h	r	h
b	a	k	r
f	p	n	d
o	p	l	j
h	y	g	c
2열에 적힌 글자를 세로로 읽으면 happy이므로 "happy"를 return 합니다.

입출력 예 #2
예제 2번의 my_string은 m이 1이므로 세로로 "programmers"를 적는 것과 같고 따라서 1열에 적힌 글자를 세로로 읽으면 programmers입니다. 따라서 "programmers"를 return 합니다.

---

내 풀이

Java

class Solution {
    public String solution(String my_string, int m, int c) {
        String answer = "";
        
        for(int i = c - 1; i < my_string.length(); i += m){
            answer += my_string.charAt(i);
        }
        
        return answer;
    }
}

C

#include <stdio.h>
#include <stdbool.h>
#include <stdlib.h>

// 파라미터로 주어지는 문자열은 const로 주어집니다. 변경하려면 문자열을 복사해서 사용하세요.
char* solution(const char* my_string, int m, int c) {
    // return 값은 malloc 등 동적 할당을 사용해주세요. 할당 길이는 상황에 맞게 변경해주세요.
    char* answer = (char*)malloc(sizeof(char) * strlen(my_string) + 1);
    
    int i, k;
    for(i = c - 1, k = 0; i < strlen(my_string); i += m){
        answer[k++] = my_string[i];
    }
    answer[k] = '\0';
    return answer;
}

Python

def solution(my_string, m, c):
    answer = my_string[c - 1::m]
    return answer

---

https://school.programmers.co.kr/learn/courses/30/lessons/181904

프로그래머스