将一个给定字符串根据给定的行数,以从上往下、从左到右进行 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 Solution:
def convert(self, s: str, numRows: int) -> str:
n = min(numRows, len(s)) # 字母排列的行数
list_str = [[] for _ in range(n)] # 创建n个空列表
if n == 1: # 如果只有一行,返回原字符串
return s
direction = 0 # 字母排列的方向。0:下 1:右上
i, j = 0, 0 # 行号和列号
# 将每个字符Z字排列
for m in s:
list_str[i].append(m)
if direction == 0: # 向下排列
if i < n-1: # 不是最后一排
i += 1
else: # 最后一排,接下来要向右上排列
i -= 1
direction = 1
elif direction == 1: # 向右上排列
if i > 0: # 不是第一排
i -= 1
list_str[i].append("")
else: # 第一排,接下来向下排序
i += 1
direction = 0
# 将列表转换成新的字符串
str_new = ""
for i in range(n):
str_new += "".join(list_str[i])
return str_new
在解答这道题的时候,逻辑上的调整就不写出来了,对于Python的使用出现语法问题:
- 一开始想给把字符串的字母填到列表的时候,使用了list_str[i][j]的写法,完全当成二维数组在处理,结果报错列表索引超出范围。
- 在把列表转换成字符串的时候,一开始写成str_new = “”.join(list_str),报错期待字符串,结果出现列表。