11716 - Digital Fortress

#include<stdio.h>
#include<math.h>
#define M 10010

int main()
{
    char cipher[M],msg[M];
    int t,i,j,dim,s;
    //freopen("in.txt","r",stdin);

    scanf("%d\n",&t);
    while(t--)
    {
        gets(cipher);
        s=0;
        for(s=0;cipher[s];s++);

        dim=(int)sqrt(s);
        if(dim*dim<s)
        {
            printf("INVALID\n");
            continue;
        }

        for(i=0;cipher[i];i++)
        {
            j=(i%dim)*dim+i/dim;
            msg[j]=cipher[i];
        }
        msg[i]='\0';

        printf("%s\n",msg);
    }
    return 0;
}