Leetcode 6.Z字形变换题解

Leetcode 6.Z字形变换题解

题目链接：

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

题目描述

将一个给定字符串根据给定的行数，以从上往下、从左到右进行 Z 字形排列。
比如输入字符串为 “LEETCODEISHIRING” 行数为 3 时，排列如下：

之后，你的输出需要从左往右逐行读取，产生出一个新的字符串，比如：“LCIRETOESIIGEDHN”。

思路

这道题很明显是一道找规律的题目，首先看样例输出的第一行，每个字符在原字符串中的位置分别是0，4，8，12，每个字符间都差4位即2*(行数-1)。然后再往下看，虽然第二行的字符多了很多，但是e,o,s,i这四个字符之间依然只是差了四位，只不过每个字符间又多了一个字符，而多的字符的位置也与当前行数是相关的，很容易看出其规律。因此按行遍历z形字符串即可得到问题解。

代码

class Solution(object):
    def convert(self, s, numRows):
        """
        :type s: str
        :type numRows: int
        :rtype: str
        """
        if numRows == 1:
            return s
        res = ''
        #按行遍历
        for i in range(numRows):
            now = i
            while now < len(s):
                res += s[now]
                tmp = 2*(numRows-1)-2*i
                if 0<i<numRows-1 and now+tmp < len(s):
                    res += s[now+tmp]
                now += 2*(numRows - 1)
        return res