1031 Hello World for U （20 分)

常规代码

#include <iostream>
#include <string>
using namespace std;
int main() {
	string S;
	cin >> S;
	int Len = S.size();
	int VLen = (Len + 1) / 3;
	char U[40][40];
	for (int i = 0; i < 40; i++)
		for (int j = 0; j < 40; j++) U[i][j] = ' ';
	int i = 0;
	if ((Len + 1) % 3 == 0) {
		for (int row = 0; row < VLen; row++) {
			if (row == VLen - 1) {
				for (int k = 0; k <= VLen; k++)
					U[row][k] = S[i++];
			}
			else U[row][0] = S[i++];
		}
		for (int row = VLen - 2; row >= 0; row--)
			U[row][VLen] = S[i++];
		int p, q;
		for (p = 0; p < VLen; p++) {
			for (q = 0; q <= VLen; q++) {
				if (q == VLen) cout << U[p][q] << endl;
				else cout << U[p][q];
			}
		}
	}
	else if ((Len + 1) % 3 == 1) {
		for (int row = 0; row < VLen; row++) {
			if (row == VLen - 1)
				for (int k = 0; k <= VLen + 1; k++) U[row][k] = S[i++];
			else U[row][0] = S[i++];
		}
		for (int row = VLen - 2; row >= 0; row--) U[row][VLen + 1] = S[i++];
		int p, q;
		for (p = 0; p < VLen; p++) {
			for (q = 0; q <= VLen + 1; q++) {
				if (q == VLen + 1) cout << U[p][q] << endl;
				else cout << U[p][q];
			}
		}
	}
	else if ((Len + 1) % 3 == 2) {
		for (int row = 0; row <= VLen; row++) {
			if (row == VLen) for (int k = 0; k <= VLen; k++) U[row][k] = S[i++];
			else U[row][0] = S[i++];
		}
		for (int row = VLen - 1; row >= 0; row--) U[row][VLen] = S[i++];
		int p, q;
		for (p = 0; p <= VLen; p++) {
			for (q = 0; q <= VLen; q++) {
				if (q == VLen) cout << U[p][q] << endl;
				else cout << U[p][q];
			}
		}
	}
	return 0;
}

优化代码

#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
int main() {
	char S[100];
	scanf("%s", S);
	int Len = strlen(S);
	int n1 = (Len + 2) / 3, n3 = n1, n2 = Len + 2 - n1 * 2;
	char U[40][40];
	for (int i = 1; i <= n1; i++)
		for (int j = 1; j <= n2; j++) U[i][j] = ' ';
	int pos = 0;
	for (int i = 1; i <= n1; i++) U[i][1] = S[pos++];
	for (int j = 2; j <= n2; j++) U[n1][j] = S[pos++];
	for (int k = n3 - 1; k > 0; k--) U[k][n2] = S[pos++];
	for (int i = 1; i <= n1; i++) {
		for (int j = 1; j <= n2; j++) cout << U[i][j];
		cout << endl;
	}
	return 0;
}