PAT（甲级）1031 Hello World for U (20point(s))

题目链接

题目链接

思路

题目大意：这道题真的把我绕进去了；输出u型，竖着着两条边上的元素个数相等，分别是n1，n3，底边元素的个数记为n2；要求 n1 = n3 < n2，并且n1的值尽可能大；
可以这样想，一共有N个字符，分别填到3个边上，所以至少要分成3份，即要用N+2除3，注意有两个元素重复了；N+2除3余数有三种可能：
余数为零，三条边元素个数相等；
余数为1，因为要求n1 < n2，所以要把多的这个元素放到底边，底边比侧边多一个元素；
余数为2，底边比侧边多2个元素；
没有其他可能了；
以后做图形输出这种题，还是全部放到二维数组里，最后一次性输出比较好；

代码

#include <iostream>
#include <string>
#include <vector>
#include <algorithm>
using namespace std;
const int maxn = 40;

int main()
{
     string s;
     cin >> s;
     char ans[maxn][maxn];
     //整体赋为空格
     fill(ans[0], ans[0] + maxn * maxn, ' ');
     //计算n1,n2,n3
     int n1 = (s.size() + 2) / 3, n3 = n1, n2 = s.size() + 2 -n1 - n3;
     int pos = 0;
     //赋第一列
     for(int i = 1; i < n1; i ++) ans[i][1] = s[pos++];
     //赋最后一行
     for(int i = 1; i <= n2; i ++) ans[n1][i] = s[pos++];
     //赋最后一列
     for(int i = n3 - 1; i >= 0; i --) ans[i][n2] = s[pos++];
     //输出图形
     for(int i = 1; i <= n1; i ++){
          for(int j = 1; j <= n2; j ++){
               printf("%c", ans[i][j]);
          }
          printf("\n");
     }
     system("pause");
     return 0;
}