将一个给定字符串根据给定的行数,以从上往下、从左到右进行 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
来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/zigzag-conversion
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
class Soultion :
def convert(self,s:str,numRows:int)->str:
if numRows<2:return s
count = 0
flag=1
res = ['' for i in range(numRows)]
for i in s:
res[count]+=i
count +=flag
if count == 0 or count ==numRows-1:flag=-flag
return "".join(res)
测试:
c = Soultion()
print(c.convert('人生苦短,我学python',3))
小结:巧妙的运用了flag改变了赋值的方向