[Level.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

문제 해결

def solution(my_string, m, c):
    return ''.join([my_string[m*i:m*(i+1)][c-1] for i in range(len(my_string)//m)])

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

 

주석: 솔직히 첫번째 코드 짜고 천재가 했는데 2번째 코드 보자마자 벽 느껴버림...