[HackerRank] Encryption

[문제 링크]

[입력]

string s: 암호화할 문자열

[출력]

string: 암호화한 문자열

[코드]

def encryption(s):
    # Write your code here
    row = 0 
    col = 0
    conv = list()
    result = ''
    for i in range(1,10):
        if(i * i <= len(s) and (i+1)*(i+1)>=len(s)):
            if(i*i == len(s)):
                row = i
                col = i
            else:
                row = i
                col = i+1
            break
    if(row*col<len(s)):
        row += 1
        
    for idx in range(row):
        conv.append(s[idx*col:(idx+1)*col])
    
    for i in range(col):
        for j in range(row):
            if((col*j)+i< len(s)):
                result+= conv[j][i]
        result += ' '
    return result