天梯练习 7-4 古风排版

#include <iostream>
#include <cstring>

using namespace std;

char a[110][110];//输出的答案

int main(){
    int k;
    cin >> k;
    getchar();//读入int后不吞掉回车
    
    string s;
    getline(cin, s);
    int n = s.size() / k;
    if(s.size() % k != 0) n ++ ;
    
    for(int i = 1; i <= k; i ++ ){
        for(int j = 1; j <= n; j ++ ){
            int u = (n - j) * k + i;//重点

            if(u > s.size()) a[i][j] = ' ';
            else a[i][j] = s[u - 1];

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