Z字形变换
题目
将一个给定字符串
s
s
s 根据给定的行数
n
u
m
R
o
w
s
numRows
numRows ,以从上往下、从左到右进行 Z 字形排列。
比如输入字符串为 "PAYPALISHIRING"
行数为 3 时,排列如下:
之后,你的输出需要从左往右逐行读取,产生出一个新的字符串,比如:“PAHNAPLSIIGYIR”。
请你实现这个将字符串进行指定行数变换的函数:
string convert(string s, int numRows);
示例
方法一:模拟变换过程
分析题目,将Z字形看作一个二维矩阵,变换的过程简单描述为:将字符串中的字符依次填入二维矩阵中,填入顺序为从[0][0]开始往下,填入 numRows 行,然后按照 [行数-1][列数+1] 每次填入一个字符,直到 行数=0,然后再依次往下填入,填入 numRows 行后再次按照 [行数-1][列数+1] 每次填入一个字符,循环往复直到字符串中的字符全部填入,最终按照行读取的方式整理出最终的字符串。
实现方式
使用一个Vector容器来存储每一行的元素,最终的Z字形一共 numRows 行,所以先初始化一个含有 numRows 个元素的Vector,按照Z字形变换的顺序,依次将第 i 行的字符放入到 Vector 中的第 i 个元素中,最终将Vector中每个元素中的字符串拼接成最终结果。
设置一个索引 vIndex 在[0,numRows-1]之间来回移动,表示当前字符对应在二维矩阵中的行数,字符串中的字符依次加入到 Vector中的第 vIndex 个元素中。
代码实现
class Solution {
public:
string convert(string s, int numRows) {
//用来遍历字符串 s 的索引
int index = 0;
int size = s.length();
if (numRows == 1 || numRows >= size) {
cout << s << endl;
return 0;
}
vector<string> nRes(numRows, "");
//vIndex用来指向当前元素对应在二维矩阵的行数,同样是Vector的元素索引
int vIndex = 0;
//判断当前行数索引应该递增还是递减
bool add = true;
while (index < size)
{
//从第0行往下依次填入 numRows 个元素
while (index < size && add) {
nRes[vIndex] += s[index];
vIndex++;
index++;
if (vIndex == numRows) {
add = false;
}
}
vIndex -= 2;
//按照 [行数-1][列数+1] 每次填入一个字符,直到 行数=0
while (!add && index < size) {
nRes[vIndex] += s[index];
vIndex--;
index++;
if (vIndex == -1) {
add = true;
}
}
vIndex += 2;
}
string res = "";
for (string var : nRes)
{
res += var;
}
return res;
};
看了力扣上的题解,也是找到了变换的规律,并使用数学公式表示出来,思路类似,都可以作为参考。