字符串字符數組 PTA 7-10 古風排版

7-10 古風排版 (20分)

中國的古人寫文字，是從右向左豎向排版的。本題就請你編寫程序，把一段文字按古風排版。

輸入格式：
輸入在第一行給出一個正整數N（<100），是每一列的字符數。第二行給出一個長度不超過1000的非空字符串，以回車結束。

輸出格式：
按古風格式排版給定的字符串，每列N個字符（除了最後一列可能不足N個）。

輸入樣例：

4
This is a test case

輸出樣例：

asa T
st ih
e tsi
 ce s

思路：
根據N想象排列好的字符矩陣，注意最後字符不夠要用空格補齊
代碼：

#include <iostream>
#include <string>
#include <cstdio>
using namespace std;
int main() {
    char c[1001][1001];
    int n;
    string p;
    cin>>n;
    getchar();
    getline(cin,p);
    int x;
    if ((p.size()-1)%n == 0)x=(p.size()-1)/n;
    else x = (p.size()-1)/n+1;
    for (int i = 0; i < n; ++i) {
        for (int j = 0; j <= x; ++j) {
            if(j*n+i > p.size()-1)
                {c[i][j]=' ';}
            else c[i][j]=p[j*n+i];
        }
    }
    for (int k = 0; k < n; ++k) {
        if (n>1)
        {for (int i = x-1; i >= 0; --i) {
            cout<<c[k][i];
        }
        if(k<n-1)
        cout<<endl;} else{
            for (int i = x; i >= 0; --i) {
                cout<<c[k][i];
            }
            if(k<n-1)
                cout<<endl;
        }
    }
    return 0;
}