PAT A1031

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

实现代码如下：

#include <iostream>
using namespace std;
int main(){
    string s;
    cin>>s;
    int i;
    for(i = 0; ; i++){
        if((s.length() - 2 * i) <= i)
            break;
    }
    int n = s.length() - 2 * i;
    for(int j = 0, l = s.length() - 1; j < i - 1; j++, l--){
        cout<<s[j];
        for(int o = 0; o < n; o++)
            cout<<" ";
        cout<<s[l];
        cout<<endl;
    }
    for(int j = i - 1; j < i + n + 1; j++){
        cout<<s[j];
    }
}