N字形变换
将一个给定字符串 s 根据给定的行数 numRows ,以从上往下、从左到右进行 Z 字形排列。
比如输入字符串为 "PAYPALISHIRING" 行数为 3 时,排列如下:
P A H N
A P L S I I G
Y I R
之后,你的输出需要从左往右逐行读取,产生出一个新的字符串,比如:"PAHNAPLSIIGYIR"。
最优解决方案
根据给定的行数 numRows,创建一个列表,每个元素是一个字符串,表示一行的字符。然后遍历给定的字符串 s,用一个变量 curRow 记录当前所在的行号,用一个变量 goingDown 记录当前是向下还是向上移动。如果 curRow 等于 0 或者 numRows - 1,就改变 goingDown 的方向。然后把 s 中的当前字符追加到列表中对应的元素中,然后更新 curRow 的值。最后把列表中的所有元素拼接起来就是结果。
例如,输入字符串为 “PAYPALISHIRING” 行数为 3 时:
- 初始化列表为 [“”, “”, “”]
- 遍历 s 中的第一个字符 P,curRow = 0, goingDown = True
- 把 P 追加到列表中第一个元素中 [“P”, “”, “”]
- 更新 curRow = 1
- 遍历 s 中的第二个字符 A,curRow = 1, goingDown = True
- 把 A 追加到列表中第二个元素中 [“P”, “A”, “”]
- 更新 curRow = 2
- 遍历 s 中的第三个字符 Y,curRow = 2, goingDown = True
- 把 Y 追加到列表中第三个元素中 [“P”, “A”, “Y”]
- 更新 curRow = 1, goingDown = False
- 遍历 s 中的第四个字符 P,curRow = 1, goingDown = False
- 把 P 追加到列表中第二个元素中 [“P”, “AP”, “Y”]
- 更新 curRow = 0
- …以此类推…
- 最后得到列表为 [“PAHN”, “APLSIIG”, “YIR”]
- 拼接起来得到结果为 “PAHNAPLSIIGYIR”
- 这种方法的时间复杂度和空间复杂度都是 O(n),其中 n 是字符串 s 的长度