将一个给定字符串根据给定的行数,以从上往下、从左到右进行 Z 字形排列。
比如输入字符串为 “LEETCODEISHIRING” 行数为 3 时,排列如下:
L C I R
E T O E S H 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
以下两个程序的方法大体是一样的:按行排序
通过从左向右迭代字符串,我们可以轻松地确定字符位于 Z 字形图案中的哪一行。
使用min(numRows,len(s))
个列表来表示 Z 字形图案中的非空行。
从左到右迭代s,将每个字符添加到合适的行。可以使用当前行和当前方向这两个变量对合适的行进行跟踪。
只有当我们向上移动到最上面的行或向下移动到最下面的行时,当前方向才会发生改变。
遍历原始字符串,用currow变量记录所处的行,即让currow变换为0,1,2,1,0,1,2,1,0,1,2,1,0,1…
(如果numRows是4行currow就是0,1,2,3,2,1,0,1,2,3,2,1… )
同时初始化一个长度为3,元素为字符串的列表rows。
每到currow行,就把遍历到的字符s加入到rows相应元素,
如第0行,rows[0]+=s,第1行,rows[1]+=s … 最后再把rows按顺序输出就好了。
class Solution(object):
def convert(self, s, numRows):
"""
:type s: str
:type numRows: int
:rtype: str
"""
if numRows == 0 or not s:return ""
if numRows == 1:return s
str1 = ""
str2 = ""
size = len(s)
if numRows == 2:
for i in range(size):
if i % 2 == 0:
str1 += s[i]
else:
str2 += s[i]
return str1+str2
cur_row = 0
s_list = ["" for i in range(min(size,numRows))]
Flag = False
#按行排序
for i in range(size):
s_list[cur_row] += s[i]
if cur_row == 0 or cur_row == numRows - 1:
Flag = not Flag
if Flag == True:
cur_row += 1
else:
cur_row -= 1
temp = ""
for j in range(len(s_list)):
temp += s_list[j]
return temp
class Solution:
def convert(self, s, numRows) :
if not s:
return ""
if numRows == 1:return s
s_Rows = [""] * numRows
i = 0
n = len(s)
while i < n:
for j in range(numRows):
if i < n:
s_Rows[j] += s[i]
i += 1
for j in range(numRows-2,0,-1):
if i < n:
s_Rows[j] += s[i]
i += 1
return "".join(s_Rows)