LeetCodeEasy-【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、找规律、每次需要输出numRows形状的每行字母，故循环numRows次，每次取余即可，对于中间的字符取余时需要例外判断：那V形状还计算，长度为 2*n-2, 再减去余数，得到的即为所在行数 。

class Solution:
    def convert(self, s: str, numRows: int) -> str:
        if numRows == 1:
            return s
        res = ''
        for i in range(numRows):
            for j in range(len(s)):
                if j % (numRows*2 - 2) == i:
                    # print(s[j])
                    res += s[j]
                else:
                    if 2*numRows - 2 - (j % (numRows*2 - 2)) == i:
                        res += s[j]  
        return res

改进版本：

class Solution:
    def convert(self, s: str, numRows: int) -> str:
        if numRows == 1:
            return s
        res = ''
        for i in range(numRows):
            for j in range(i, len(s), numRows*2-2):
                res += s[j]
                if numRows > 2 and i != 0 and i != numRows-1 and j + (numRows - i - 1)*2 < len(s):
                    res += s[j + (numRows - i - 1)*2]
                    # print(i, j, j + (numRows - i - 1)*2)
        return res

思路二、

class Solution:
    def convert(self, s: str, numRows: int) -> str:
        if numRows < 2: return s
        res = ["" for _ in range(numRows)]
        i, flag = 0, -1
        for c in s:
            res[i] += c
            if i == 0 or i == numRows - 1: flag = -flag
            i += flag
        return "".join(res)