PAT 大区赛 1-7古风排版

/**
L1-039. 古风排版
中国的古人写文字，是从右向左竖向排版的。本题就请你编写程序，
把一段文字按古风排版。


输入格式：


输入在第一行给出一个正整数N（<100），是每一列的字符数。
第二行给出一个长度不超过1000的非空字符串，以回车结束。


输出格式：


按古风格式排版给定的字符串，每列N个字符（除了最后一列可能不足N个）


输入样例：
4
This is a test case
输出样例：
asa T
st ih
e tsi
 ce s
*/

#include<iostream>
using namespace std;
int main(){
int n;
cin>>n;
getchar();
char s[1000];
char c;
int count=0,i,j,k;
while((c=getchar())!='\n'){
s[count]=c;
count++;
} 
int row=0,col=0;
int m=count/n;
if(count%n) m++;
char array[n][m];
for(i=0;i<n;i++){
for(j=0;j<m;j++){
array[i][j]=' ';
}
}
for(k=0;k<count;k++){
row=k%n;
col=k/n;
array[row][col]=s[k];
}
for(i=0;i<n;i++){
for(k=m-1;k>=0;k--){
cout<<array[i][k];
} 
cout<<endl;
}
return 0;
}