【Leetcode】6. Z 字形变换（python版本）【单list】

题目：

链接:https://leetcode-cn.com/problems/zigzag-conversion/

将一个给定字符串根据给定的行数，以从上往下、从左到右进行 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

思路：

利用一个numRows大小的列表，列表中的元素是字符串，列表中第n个字符串表示的是变幻后的第n行的字符串。
一次遍历原字符串，将每个字符添加到对应的行中（即追加到列表中对应元素字符串后），第0 行和第numRows行换向。

代码：

class Solution:
    def convert(self, s: str, numRows: int) -> str:
        if numRows == 1: 
            return s
        stra = [''] * numRows
        row = 0
        for i in s:
            stra[row] = stra[row] + i
            if row == 0:
                to = 1
            elif row == numRows-1:
                to = -1
            row = row + to
        re = ''
        for i in range(numRows):
            re = re + stra[i]
        return re