PTA团体程序设计天梯赛-练习集L1-039 古风排版

L1-039 古风排版

题目要求

中国的古人写文字，是从右向左竖向排版的。本题就请你编写程序，把一段文字按古风排版。
输入格式:
输入在第一行给出一个正整数N（<100），是每一列的字符数。第二行给出一个长度不超过1000的非空字符串，以回车结束。
输出格式:
按古风格式排版给定的字符串，每列N个字符（除了最后一列可能不足N个）。
输入样例:

4
This is a test case

输出样例:

asa T
st ih
e tsi
 ce s

代码如下：

#include<iostream>
#include<algorithm>
#include<string>
using namespace std;
int main()
{
    int n;
    string str;
    cin>>n;
    getchar();
    getline(cin,str);
    int col=(str.size()+n-1)/n;
    char s[col][n];
    fill(s[0],s[0]+col*n,' ');
    int flag=0;
    for(int i=0;i<col;++i)        //得到矩阵
    {
        for(int j=0;j<n;++j)
        {
            s[i][j]=str[flag++];
            if(flag==str.size())    //此处一定要判断flag有没有超过字符总数目
                break;
        }
              
    }                          
    for(int i=0;i<n;++i)       
    {
        for(int j=col-1;j>=0;--j)

            cout<<s[j][i];
        cout<<endl;
    }
    return 0;
}