PTA L1-039 古风排版 (20 分)

L1-039 古风排版 (20 分)

#include <bits/stdc++.h>
using namespace std;
int main()
{
	string s;
	int n;
	cin>>n;
	getchar();
	getline(cin,s);
	int lie;
	if(s.length()%n==0){//一定要注意是不是刚刚好不然测试点1，3会出错 格式错误
		lie=s.length()/n;
	}
	else {
		lie=s.length()/n+1;
	}
	int i,j;
	char a[n][lie];
	int count=0;
	for(i=lie-1;i>=0;i--){
		for (j=0;j<n;j++){
			if(count<s.length()){
				a[j][i]=s[count];
				count++;
			}
			else {
				a[j][i]=' ';
			}
		}
	}
	for (i=0;i<n;i++){
		for (j=0;j<lie;j++){
			cout<<a[i][j];
		}
		cout<<endl;
	}
	return 0;
}