题目描述:
将一个给定字符串根据给定的行数,以从上往下、从左到右进行 Z 字形排列。
比如输入字符串为 “LEETCODEISHIRING” 行数为 3 时,排列如下:
L C I R
E T O E S I I G
E D H N
实例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
解题思路:
1)建立numRows个列表ListA[i],每个列表用来存储相应行的字符串。
2)如果只有一个字符(numRows==1)直接return。
3)设置flag参数,初始为False,作为是否继续按列表的正向存储的标志。
4)遍历字符串,没当index为numRows-1的整数倍的时候,flag反制。之后根据flag的情况按顺序存储字符到相应的列表。
5)将列表组合起来并输出。
T T T
T F T F T
T F T F T
F F F
可以看到,在flag为True的时候,ListA[i]从0-3行正向存储,在flag为False时,ListA[i]从4-2行反向存储,故,程序可以如下设计:
class Solution:
def convert(self, s: str, numRows: int) -> str:
ListA= [[]for i in range(numRows)]
if numRows==1:
return s
flag= False
for i,string in enumerate(s):
j = i%(numRows-1)
if j == 0:
flag = not flag
if flag:
ListA[j].append(string)
else:
ListA[numRows - 1 - j].append(string)
cache = [''.join(ListA[i]) for i in range(numRows)]
return ''.join(cache)
参考别人写的,自己重新理解了一遍