public static String encryption(String s) {
    // Write your code here
    s = s.replaceAll(" ", "");
    int sqr = (int)Math.ceil(Math.sqrt(s.length()));
    String[] result = new String[sqr];
    for(int i = 0; i < sqr; i++){
        for(int j = i; j < s.length();j = j+sqr){
            char c = s.charAt(j);
            result[i] = j >= sqr ?  result[i] + c : "" + c;
        }
    }
    return String.join(" ", result);
    }

def encryption(s):
    s = s.replace(' ', '')
    sqr = len(s)**(1/2)
    rows, cols = round(sqr), math.ceil(sqr)
    grid = []
    ptr = 0
    for i in range(rows):
        grid.append(s[ptr: ptr+cols])
        ptr += cols
    
    output = []
    
    for i in range(cols):
        word = ''
        for j in range(rows):
            try:
                word += grid[j][i]
            except Exception:
                continue
        output.append(word)
            
        
    return ' '.join(output)

from math import sqrt
from math import floor
from math import ceil

def encryption(s):
    s = s.replace(" ", "")
    len_s = len(s)
    L = sqrt(len_s)
    r = floor(L)
    c = ceil(L)
    first = []
    for i in range(r+1):
        try:
            first.append(s[:c])
            s = s[c:]
        except:
            pass
    count = 0
    string = ""
    final = []
    for i in range(c):

        for i in first:

            try:
                string += i[count]

            except:
                pass
        count += 1
        final.append(string)
        string = ""

    return ' '.join(final)

    a=math.floor(math.sqrt(len(s)))
    b=math.ceil(math.sqrt(len(s)))
    if not (a*b >= len(s)):
        a+=1
    l=[""]*a
    k=0
    for i in range(a):
        for j in range(b):
            if len(s)>k:
                l[i]+=s[k]
                k+=1
    res=""
    
    for i in range(len(l[0])):
        for j in range(len(l)):
            try:
                res+=l[j][i]
            except:pass
        res+=" "
    return res