但你必须不断尝试,每天表现你最好的一面。

题目描述:

N 字形变换

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

比如输入字符串为 "PAYPALISHIRING" 行数为 3 时,排列如下:

1
2
3
P   A   H   N
A P L S I I G
Y I R

之后,你的输出需要从左往右逐行读取,产生出一个新的字符串,比如:"PAHNAPLSIIGYIR"

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

1
string convert(string s, int numRows);

解题代码:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
class Solution {
public:
string convert(string s, int numRows) {
if(numRows < 2) return s;
int i = 0;
int flag = -1;
vector<string> rows(numRows);
for(char c : s){
if(i == 0 || i == numRows - 1){
flag = -flag;
}
rows[i] += c;
i += flag;
}
string ans;
for(string row : rows){
ans += row;
}
return ans;
}
};

image-20231023163046137

解读与收获:

本题算法模拟了题目中的”Z“形成的过程,设置下标i和标志位flag,从头遍历字符串s,期间往字符串数组rows中填充字符,其中i下标模拟了Z的每一层,如果i等于0或是层数-1,说明到了转折点,此时改变flag为-flag,i的值便会往相反方向增长,这就实现了类似于反复扫描的过程,”扫描“结束后,字符串数组每一个元素对应的就是Z的每一层。