【题解】【AcWing】1515. U 形 Hello World

1515. U 形 Hello World

原题传送：AcWing 1515. U 形 Hello World

给定一个长度为 $N$ 的字符串，请你将它以 $U$ 形输出。

例如，helloworld可以输出为：

h  d
e  l
l  r
lowo

也就是说，必须按照原始顺序输出字符，左垂直线自上而下共有 $n_1$ 个字符，底部行从左到右共有 $n_2$ 个字符，右垂直线自下而上共有 $n_3$ 个字符。

另外，必须满足 n 1 = n 3 = m a x { k ∣ k ≤ n 2 对 于 所 有 3 ≤ n 2 ≤ N } n_1=n_3=max \lbrace k|k \le n_2对于所有3 \le n_2 \le N\rbrace n1​=n3​=max{k∣k≤n2​对于所有3≤n2​≤N} 以及 $n_1+n_2+n_3-2=N$ 。

对于helloworld， $n_1=n_2=n_3=4$ 。

输入格式

包含一个长度不少于 $5$ 且不超过 $80$ 的字符串，字符串内不含空格。

输出格式

输出转化为 $U$ 形后的图案。

输入样例：

helloworld!

输出样例：

h   !
e   d
l   l
lowor

思路：

通过题中所给的条件计算出n1、n2，然后将字符填充到对应位置。

题解：

#include<bits/stdc++.h>

using namespace std;

int main()
{	
	string s;
	vector<vector<char>> res;
	cin >> s;
	
	int sz = s.length();
	int n1 = (sz + 2) / 3, n2 = n1 + (sz + 2) % 3;
	for(int i = 0; i < n1; i++)
	{
		vector<char> v;
		for(int j = 0; j < n2; j++)
		{
			if(i != n1 - 1)
			{
				if(!j)
					v.push_back(s[i]);
				else if(j == n2 - 1)
					v.push_back(s[sz - i -1]);
				else
					v.push_back(' ');
			}
			else
			{
				v.push_back(s[i +j]);
			}
		}
		res.push_back(v);
	}
	
	for(int i = 0; i < res.size(); i++)
	{
		for(int j = 0; j < res[i].size(); j++)
		{
			cout << res[i][j];
		}
		cout << endl;
	}

	return 0;
}