백준 2999 비밀 이메일

https://www.acmicpc.net/problem/2999

2999번: 비밀 이메일

 

def cal(n):
    r,c=0,0
    for i in range(1, n):
        if n%i==0 and i<=n//i:
            r,c= i,n//i
    return r,c
def sol(s,r,c):
    answer=[]
    for i in range(c):
        tmp=''
        for j in range(r):
            tmp+=s[r*i+j]
        answer.append(tmp)
    answer_string=''
    for i in range(r):
        for j in range(c):
            answer_string+=answer[j][i]
    return answer_string
s=input()
r,c=cal(len(s))
print(sol(s,r,c))

cal 함수는 r,c를 조건에 맞게 구하는 함수입니다.
(Cal fucntion is to calculate r, c in this problem)

sol 함수는
크게 두가지 내용인데
첫번째는 메세지를 행렬에 맞게 쪼개는 것, 그리고 두번째 문제 내용처럼 위에서 아래로 읽는것
(sol fucntion is Main function, First of all this means split the message into matrix, and read like the way which is like in problem Up and downs)