프로그래머스 181904번 세로 읽기 - Python, Java
문제 설명
문자열 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 합니다.
코드 구현
Python
def solution(my_string, m, c):
if m == 1:
return my_string
else:
answer = ""
for i in range(c - 1, len(my_string), m):
answer += my_string[i]
return answer
Java
class Solution {
public String solution(String my_string, int m, int c) {
if (m == 1) {
return my_string;
} else {
StringBuilder answer = new StringBuilder();
for (int i = c - 1; i < my_string.length(); i += m) {
answer.append(my_string.charAt(i));
}
return answer.toString();
}
}
}