2018.8.5拼多多笔试题1———正方形字符

输入描述：输入一行，包含4K(K为整数，1<=k<=10)个小写字母
输出描述：输出k+1行，按上面的规则输出正方形，正方形内部用空格填充
示例输入：abcdefghijklmnop
输出：
abcde
p f
o g
n h
mlkji

#include <iostream>
#include <vector>
using namespace std;

int main()
{
    vector<char> input;
    char a;
    int count = 0;
    while(cin>>a)
    {
        input.push_back(a);
    }
    count=input.size();
    int k=count/4;
    for(int i=0;i<k+1;i++)
        cout<<input[i];
    cout<<endl;
    for(int j=0;j<k-1;j++)
    {
        cout<<input[4*k-1-j];
        for(int space=0;space<k-1;space++)
            cout<<' ';
        cout<<input[k+1+j];
        cout<<endl;
    }
    for(int i=3*k;i>=2*k;i--)
        cout<<input[i];
    return 0;
}