解题思路-leetcode第六题:Z字形变换
题目描述:将一个给定字符串根据给定的行数,以从上往下、从左到右进行 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
看到题目首先找规律,发现当行数为3时,字符串行坐标为1232123212321,行数为4时,行坐标为1234321234321,故设计了一个变量x表示行坐标,设计一个变量type表示行数增大还是减小。代码一开始根据行数动态定义了一个列表,列表内元素均为空字符串,然后定义了一个字符串变量用于存储返回的字符串。然后是代码主体部分,先排除特殊情况,然后通过for循环遍历字符串,将字符根据行数赋值给相应的子字符串,当行数增加到最大时,改变type的值,行坐标逐渐减小,依次类推,直至遍历完字符串,最后返回子字符串相加的和。代码如下:
class Solution:
def convert(self, s: str, numRows: int) -> str:
x, type = -1, 1
r = [""for _ in range(numRows)]
nuws = ""
if numRows <= 1:
return s
for a in s:
if type == 1:
x += 1
r[x] += a
if x == numRows-1:
type = -1
else:
x -= 1
r[x] += a
if x == 0:
type = 1
for i in range(numRows):
nuws += r[i]
return nuws
提交后,通过。