LeetCode 6. Z 字形变换【Python 数学规律法】

本文发布且更新于个人博客：https://www.xerrors.fun/leetcode/zigzag-conversion/

1. 题目

题目链接：https://leetcode-cn.com/problems/zigzag-conversion/submissions/

2. 分析思路

第一部分是自己的分析思路，后一部分是参考资料之后的完善的思路，记录自己的思考过程的同时记录优秀的解法，把这个题目弄明白

思考过程

这个第一反应是找规律，所以使用的是数学规律法。根据分析可以发现，每一行输出的数据对应的位置是有规律可以找的；

为了减少对于字符串长度的判断，首先把字符串填充为符合上述结构的长度，填充部分为字符 0， 当遍历结束之后再把结果里面的 0 删除；

完善解法

在题解中发现主要有两种解法，一种是模拟输出方法，将每一行的数据保存到一个数组/字符串里面，最后把这些行的数据放在一个变量里面作为结果，也很容易理解。来源：Z 字形变换 c++实现两种解法 模拟法和数学规律法 图示讲解

代码及结果

class Solution:
    def convert(self, s: str, numRows: int) -> str:
        size = len(s)
        if size <= numRows or numRows == 1:
            return s

        k = (size - numRows) // ((numRows - 1) * 2)

        s = s + "0"*2*(numRows-1)
        r = ""
        for i in range(numRows):
            r += s[i]
            for j in range(1, k+2):
                if i == 0:
                    r += s[(2*j)*(numRows-1)]
                elif i == numRows-1:
                    r += s[(2*j+1)*(numRows-1)]
                else:
                    r += s[(2*j)*(numRows-1) - i] + s[(2*j)*(numRows-1) + i]
        return r.replace('0', '')

参考资料

[1] Z 字形变换 c++实现两种解法 模拟法和数学规律法 图示讲解