题目:
将一个给定字符串根据给定的行数,以从上往下、从左到右进行 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
思路:
观察结构,其可分为cov=num*2-2个数字一组
1.每层构建一个字符串,对字符串进行遍历,将结果放入对应得层,合并字符串
2.取固定行数,每行按序号从str中取
思路1
def convert(self, s: str, numRows: int) -> str:
if numRows == 1:
return s
cov = numRows*2-2
l = ['']*numRows
for i in range(len(s)):
n = i%cov
if n >= numRows:
l[cov-n] +=s[i]
else:
l[n] += s[i]
return ''.join(l)
思路2
def convert(self, s: str, numRows: int) -> str:
if numRows == 1:
return s
cov = numRows*2-2
l = ''
for i in range(numRows):
j = 0
while j+i < len(s):
l+=s[i+j]
if i!=0 and j+cov-i < len(s) and i!=numRows-1:
#除第一行和最后一行,其它第i行得j+i个与cov-i+j个同处一行
l+=s[j+cov-i]
j+=cov
return l
思路1优化z形遍历
def convert(self, s, numRows):
if numRows == 1 or numRows >= len(s):
return s
L = [''] * numRows
index, step = 0, 1
for x in s:
L[index] +=x
if index == 0:
step = 1
elif index == numRows -1:
step = -1
index += step
return ''.join(L)