题目描述:
将一个给定字符串根据给定的行数,以从上往下、从左到右进行 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;
}