浙大2012上机 PAT 1031. Hello World for U (20)

Given any string of N (>=5) characters, you are asked to form the characters into the shape of U. For example, "helloworld" can be printed as:

h  d
e  l
l  r
lowo

That is, the characters must be printed in the original order, starting top-down from the left vertical line with n ₁ characters, then left to right along the bottom line with n ₂  characters, and finally bottom-up along the vertical line with n ₃  characters. And more, we would like U to be as squared as possible -- that is, it must be satisfied that n ₁  = n ₃  = max { k| k <= n ₂  for all 3 <= n ₂  <= N } with n ₁  + n ₂  + n ₃  - 2 = N.

Input Specification:

Each input file contains one test case. Each case contains one string with no less than 5 and no more than 80 characters in a line. The string contains no white space.

Output Specification:

For each test case, print the input string in the shape of U as specified in the description.

Sample Input:

helloworld!

Sample Output:

h   !
e   d
l   l
lowor

注意考虑字符串长度仅为5的情况。

代码如下：

#include<iostream>
#include<vector>
#include<algorithm>
#include<cstdio>
#include<cstring>
int main()
{
#ifdef ONLINE_JUDGE
#else
	freopen("D:\\in.txt","r",stdin);
	freopen("D:\\out.txt","w",stdout);
#endif
    char str[1000];
	scanf("%s",str);
	int len=strlen(str);
	int maxt=0;
	//for(int i=3;i<=len/2;i++)
	//当字符串长度为5时，则排版不正确
	for(int i=3;i<=len;i++)
	{
	   int t=(len-i)/2;
	   if(t>maxt&&t<i)
	   {
		   maxt=t;
	   }
	}
	int max2=len-maxt*2;
	for(int i=0;i<maxt;i++)
	{
		printf("%c",str[i]);
		for(int j=0;j<max2-2;j++)
			printf(" ");
		printf("%c\n",str[len-i-1]);
	}
	for(int i=maxt;i<maxt+max2;i++)
		printf("%c",str[i]);
	return 0;
}