【lc刷题】6 Z 字形变换 (138/300)

138/300

6. 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

思路1：
如图：

class Solution:
    def convert(self, s: str, numRows: int) -> str:
        if numRows < 2: return s
        res = ['']*numRows
        n, d = 0, -1
        for c in s:
            res[n] += c
            if n == numRows-1 or n == 0:
                d = -d
            n += d
        return  ''.join(res)

思路2:
观察规律

知识点：
Python 字符串/列表 切片slice 正负 步长 （详细举例）
python中 zip截短 zip_longest补长 的使用

class Solution:
    def convert(self, s: str, numRows: int) -> str:
        if numRows < 2: return s
        n = numRows-1
        step = n*2
        res = s[::step]

        for i in range(1,n):
            for v,w in itertools.zip_longest(s[i::step],s[step-i::step],fillvalue=''):         
                res += v+w
          
        return res + s[n::step]