力扣LeetCode刷题日记(四) Z 字形变换

将一个给定字符串根据给定的行数,以从上往下、从左到右进行 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的使用出现语法问题:

  1. 一开始想给把字符串的字母填到列表的时候,使用了list_str[i][j]的写法,完全当成二维数组在处理,结果报错列表索引超出范围。
  2. 在把列表转换成字符串的时候,一开始写成str_new = “”.join(list_str),报错期待字符串,结果出现列表。

你可能感兴趣的:(LeetCode刷题)