leetcode -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

思路1:

最初的思路是每一行作为一个字符串储存，然后通过指针的方式把字符串循环加入到每一行的字符串中，最后统一join打印。

code:

class Solution:
    def convert(self, s: str, numRows: int) -> str:
        if numRows < 2: return s
        nump = ['' for i in range(numRows)]
        i = 0
        row ,plus=0,-1
        while i<len(s):
            nump[row]+=s[i]
            i+=1
            if row ==numRows-1 or row==0: plus=-plus
            row+=plus
        return ''.join([''.join(str_) for str_ in nump])

思路2:

找到索引规律，首先打印第一行，然后打印第二行，依次类推…

一共有3个变量,总行数:numRows , 当前行:row , 第几列:i
我们可以把z字拆分成两部分
第1部分竖直部分:间距为2numRows-2
第2部分中间连接斜线部分:与左竖线间距为2(numRows-row-1)

code:

class Solution:
    def convert(self, s: str, numRows: int) -> str:
        if numRows < 2: return s

        res = ''

        for row in range(numRows):
            print(row)
            plus = -2 * row
            if row==0 or row==numRows-1:
                res += ''.join([s[i]  for i in range(row,len(s),2*numRows-2)])
            else:
                for i in range(row, len(s), 2 * numRows - 2):

                    res+=s[i]

                    if i+2*(numRows-row-1)<len(s):

                        res+=s[i+2*(numRows-row-1)]
        return res

两种方法的时间复杂度都为n(每个元素均访问1次)