LeetCode字符串-6Z字形变换

题目描述：

将一个给定字符串根据给定的行数，以从上往下、从左到右进行 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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/zigzag-conversion
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

 

一种解题思路：

分享一个思路，可能也是为什么叫做Z字形变换的原因。 将原题目中的排列方式做类似矩阵转置的操作，这时候整个排列就是类似Z字形的变换，例如：

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

所以想办法通过LEETCODEISHIRING得到转置之后的矩阵，然后只要先纵向输出再横向输出就可以了。

 

参考代码：

string convert(string s, int numRows) {
	if (numRows == 1)
		return s;

	string ans;
	vector<vector<char>> zShapeMatrix;
	vector<char> tmp;
	int numOfPeriod = s.length() / (2 * numRows - 2);
	int left = s.length() % (2 * numRows - 2);

	for (int i = 0; i < numOfPeriod; i++) {
		tmp.clear();
		// x x x
		for (int j = 0; j < numRows; j++) 
			tmp.push_back(s[i*(2 * numRows - 2) + j]);
		zShapeMatrix.push_back(tmp);
		// x x x
		//   *
		// *
		for (int j = 0; j < numRows - 2; j++) {
			tmp.clear();
			for (int k = 0; k < numRows - j - 2; k++)
				tmp.push_back('\0');
			tmp.push_back(s[i*(2 * numRows - 2) + numRows + j]);
			for (int k = 0; k < j + 1; k++)
				tmp.push_back('\0');
			zShapeMatrix.push_back(tmp);
		}
	}

	if (left == 0) {
	}
	else if (left <= numRows) {
		tmp.clear();
		for (int i = 0; i < left; i++) 
			tmp.push_back(s[(numOfPeriod)*(2 * numRows - 2) + i]);
		for (int i = 0; i < numRows - left; i++)
			tmp.push_back('\0');
		zShapeMatrix.push_back(tmp);
	}
	else {
		tmp.clear();
		for (int i = 0; i < numRows; i++)
			tmp.push_back(s[(numOfPeriod)*(2 * numRows - 2) + i]);
		zShapeMatrix.push_back(tmp);
		for (int i = 0; i < left - numRows; i++) {
			tmp.clear();
			for (int j = 0; j < numRows - i - 2; j++)
				tmp.push_back('\0');
			tmp.push_back(s[(numOfPeriod)*(2 * numRows - 2) + numRows + i]);
			for (int j = 0; j < i + 1; j++)
				tmp.push_back('\0');
			zShapeMatrix.push_back(tmp);
		}
	}

	for (int j = 0; j < numRows; j++) 
		for (int i = 0; i < zShapeMatrix.size(); i++) {
			if (zShapeMatrix[i][j] == '\0')
				continue;
			ans += zShapeMatrix[i][j];
		}
	return ans;
}