力扣第6题：Z 字形变换

问题描述：

将一个给定字符串根据给定的行数，以从上往下、从左到右进行 Z 字形排列。

比如输入字符串为 “LEETCODEISHIRING” 行数为 3 时，排列如下：

L   C   I   R
E T O E S I I G
E   D   H   N

之后，你的输出需要从左往右逐行读取，产生出一个新的字符串，比如：“LCIRETOESIIGEDHN”。

请你实现这个将字符串进行指定行数变换的函数：

string convert(string s, int numRows);

示例 1:

输入: s = "LEETCODEISHIRING", numRows = 3
输出: "LCIRETOESIIGEDHN"

示例 2:

输入: s = "LEETCODEISHIRING", numRows = 4
输出: "LDREOEIIECIHNTSG"

解释:

L     D     R
E   O E   I I
E C   I H   N
T     S     G

我的解答：

c++实现

class Solution {
public:
    string convert(string s, int numRows) {
        
     
	if (s.length() <= numRows|| numRows<=1)
	{
		return s;
	}
	
	vector<char> temp;

	for (int i = 0; i < numRows; i++)
	{
		int now = i;
		while (now< s.length()) {
			
			temp.push_back(s[now]);
			if (i==0||i== numRows -1)
			{
				now = now + (numRows - 1) * 2;
				continue;
			}
			now = (now + (numRows - 1) * 2) - i * 2;
			if (now >= s.length())
			{
					break;
			}
			temp.push_back(s[now]);
			now = now + i * 2;
		}
	}
	  //此处还能进一步优化
	  string s1(temp.begin(), temp.end());
	  return s1;
    }
};

运行结果：

解析：

0        8         16
1     7  9      15 17
2   6    10   14   18
3 5      11 13
4        12

我是通过数学计算，按行获取每行的字符，写入vector< char>中，除了第一行和最后一行，都是读取一次，剩下的都是读取两次，第一行的每个字符相差为（行数-1）*2，中间的每行，z的竖下位置为，(当前位置+ (总行数 - 1) * 2) - （当前行数-1） * 2。

如上图 0 和 8 相差（5 - 1）*2；2 的下一个为 6 相差 （2 + （5 - 1）2）- （3 -1） 2