Solution to the Encryption problem on Hackerrank. https://www.hackerrank.com/challenges/encryption

encription.py

#!/bin/python

import sys
from math import floor, ceil

sentence = raw_input().replace(" ", "")
size = len(sentence)
root = size ** .5

row = int(floor(root))
column = int(ceil(root))

while row * column < size:
    if row < column:
        row += 1
    else:
        column += 1

sentence += (((row * column) - len(sentence)) * " ")

matrix = []

for i in xrange(row):
    matrix.append(sentence[i * column:column * (i + 1)])
    
for i in xrange(column):
    word = ""
    for j in xrange(row):
        word += matrix[j][i]
    sys.stdout.write(word.strip() + " ")

instruction.md

An English text needs to be encrypted using the following encryption scheme. First, the spaces are removed from the text. Let be the length of this text. Then, characters are written into a grid, whose rows and columns have the following constraints:

, where is floor function and is ceil function For example, the sentence if man was meant to stay on the ground god would have given us roots after removing spaces is characters long, so it is written in the form of a grid with 7 rows and 8 columns.

ifmanwas
meanttos
tayonthe
groundgo
dwouldha
vegivenu
sroots Ensure that If multiple grids satisfy the above conditions, choose the one with the minimum area, i.e. . The encoded message is obtained by displaying the characters in a column, inserting a space, and then displaying the next column and inserting a space, and so on. For example, the encoded message for the above rectangle is:

imtgdvs fearwer mayoogo anouuio ntnnlvt wttddes aohghn sseoau

You will be given a message in English with no spaces between the words. The maximum message length can be characters. Print the encoded message.

Here are some more examples:

Sample Input:

haveaniceday Sample Output:

hae and via ecy Sample Input:

feedthedog
Sample Output:

fto ehg ee dd Sample Input:

chillout Sample Output:

clu hlt io