PTA刷题记录：L1-039 古风排版 (20分)

#include <iostream>

using namespace std;

int main() {
	int row, column;
	string str;
	
	cin >> row;
	cin.ignore();
	getline(cin, str);
	
	if (str.size() % row == 0) {
		column = str.size() / row;
	} else {
		if (str.size() <= row) {
			column = 1;
		} else {
			column = (str.size() / row) + 1;
		}
	}
		
	char mat[row][column];
	
	for (int i = 0; i < row; i++) {
		for (int j = 0; j < column; j++) {
			mat[i][j] = ' ';
		}
	}
	
	int p = 0;
	for (int j = column - 1; j > -1; j--) {
		for (int i = 0; i < row; i++) {
			if (p != str.size()) {
				mat[i][j] = str[p];
				p++;
			}
		}
	}
	
	for (int i = 0; i < row; i++) {
		for (int j = 0; j < column; j++) {
			cout << mat[i][j];
		}
		cout << endl;
	}
	
	return 0;
}