将一个给定字符串根据给定的行数,以从上往下、从左到右进行 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

题解 找出规律 然后按行读取 , 不断的获取同行下一个元素 在整个字符串上的位置。

class Solution(object):
    def convert(self, s, numRows):
        """
        :type s: str
        :type numRows: int
        :rtype: str
        """
        strLen = len(s)
        res = ""
        if strLen <= 1 or numRows == 1 :
            return s
        #规律 逐行读取
        cycleLen = 2 * numRows - 2 
        for i in range( 0 ,numRows): #循环行
            for j in range( 0 , strLen - i , cycleLen ):#第几个
                res += s[j+i]  
                if i != 0 and i != numRows -1 and j + cycleLen - i < strLen :
                    #首行尾行没有中间值 ,个数够 ,有中间值.
                    res += s[j + cycleLen - i ]
        
        return res

官方优解 每一行都创建一个字符串,找出规律 对于每一个循环内部都不外乎0到numRows行 对每个元素添加到相应的行中即可

class Solution(object):
    def convert(self, s, numRows):
        """
        :type s: str
        :type numRows: int
        :rtype: str
        """
        if numRows == 1:
            return s
        s_rows = [""] * numRows
        s_len = len(s)
        loop_len = 2*numRows-2
        for i in range(s_len):
            ref_num = i % loop_len
            if ref_num >= numRows:
                ref_num = 2*numRows-ref_num-2
            s_rows[ref_num]+=s[i]
        return "".join(s_rows)

远航__cz
28 声望2 粉丝