python LeetCode 6.Z字變換

lentrue 發(fā)布于2019-07-31 11:30 / 1806人閱讀

摘要：將一個(gè)給定字符串根據(jù)給定的行數(shù)，以從上往下從左到右進(jìn)行字形排列。請(qǐng)你實(shí)現(xiàn)這個(gè)將字符串進(jìn)行指定行數(shù)變換的函數(shù)示例輸入輸出示例輸入輸出解釋題解找出規(guī)律然后按行讀取，不斷的獲取同行下一個(gè)元素在整個(gè)字符串上的位置。

將一個(gè)給定字符串根據(jù)給定的行數(shù)，以從上往下、從左到右進(jìn)行?Z 字形排列。

比如輸入字符串為 "LEETCODEISHIRING"?行數(shù)為 3 時(shí)，排列如下：

L C I R
E T O E S I I G
E D H N
之后，你的輸出需要從左往右逐行讀取，產(chǎn)生出一個(gè)新的字符串，比如："LCIRETOESIIGEDHN"。

請(qǐng)你實(shí)現(xiàn)這個(gè)將字符串進(jìn)行指定行數(shù)變換的函數(shù)：

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

題解找出規(guī)律然后按行讀取，不斷的獲取同行下一個(gè)元素在整個(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
        #規(guī)律 逐行讀取
        cycleLen = 2 * numRows - 2 
        for i in range( 0 ,numRows): #循環(huán)行
            for j in range( 0 , strLen - i , cycleLen ):#第幾個(gè)
                res += s[j+i]  
                if i != 0 and i != numRows -1 and j + cycleLen - i < strLen :
                    #首行尾行沒(méi)有中間值 ,個(gè)數(shù)夠 ,有中間值.
                    res += s[j + cycleLen - i ]
        
        return res

官方優(yōu)解每一行都創(chuàng)建一個(gè)字符串，找出規(guī)律對(duì)于每一個(gè)循環(huán)內(nèi)部都不外乎0到numRows行對(duì)每個(gè)元素添加到相應(yīng)的行中即可

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)