L1-039 古风排版 c++

#include <bits/stdc++.h>
using namespace std;

char a[1000][1000];

int main() {
    int n;
    cin >> n;
    getchar(); // 读取并丢弃换行符
    string s;
    getline(cin, s);

    int column = s.length() % n ? s.length() / n + 1 : s.length() / n;
    int index = 0;
    for (int j = column - 1; j >= 0; j--) { // 修改此处的循环条件
        for (int i = 0; i < n; i++) {
            if (index < s.length()) // 防止越界访问
                a[i][j] = s[index++];
            else
                a[i][j] = ' '; // 如果字符串长度不足，用空格填充
        }
    }

    for (int i = 0; i < n; i++) {
        for (int j = 0; j < column; j++) {
            cout << a[i][j];
        }
        cout << endl;
    }

    return 0;
}