leetcode(6)Z字形变换——python

将一个给定字符串根据给定的行数，以从上往下、从左到右进行 Z 字形排列。

比如输入字符串为 “LEETCODEISHIRING” 行数为 3 时，排列如下：

L C I R
E T O E S H 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

以下两个程序的方法大体是一样的：按行排序

通过从左向右迭代字符串，我们可以轻松地确定字符位于 Z 字形图案中的哪一行。
使用min(numRows,len(s)) 个列表来表示 Z 字形图案中的非空行。
从左到右迭代s，将每个字符添加到合适的行。可以使用当前行和当前方向这两个变量对合适的行进行跟踪。
只有当我们向上移动到最上面的行或向下移动到最下面的行时，当前方向才会发生改变。
遍历原始字符串，用currow变量记录所处的行，即让currow变换为0,1,2,1,0,1,2,1,0,1,2,1,0,1…
（如果numRows是4行currow就是0,1,2,3,2,1,0,1,2,3,2,1… ）
同时初始化一个长度为3，元素为字符串的列表rows。
每到currow行，就把遍历到的字符s加入到rows相应元素，
如第0行，rows[0]+=s，第1行，rows[1]+=s … 最后再把rows按顺序输出就好了。

class Solution(object):
    def convert(self, s, numRows):
        """
        :type s: str
        :type numRows: int
        :rtype: str
        """
        if numRows == 0 or not s:return ""
        if numRows == 1:return s
        str1 = ""
        str2 = ""
        size = len(s)
        if numRows == 2:
            for i in range(size):
                if i % 2 == 0:
                    str1 += s[i]
                else:
                    str2 += s[i]
            return str1+str2
        cur_row = 0
        s_list = ["" for i in range(min(size,numRows))]
        Flag = False
        #按行排序
        for i in range(size):
            s_list[cur_row] += s[i]
            if cur_row == 0 or cur_row == numRows - 1:
                Flag = not Flag
            if Flag == True:
                cur_row += 1
            else:
                cur_row -= 1
        temp = ""
        for j in range(len(s_list)):
            temp += s_list[j]
            
        return temp
        

class Solution:
    def convert(self, s, numRows) :
        if not s:
            return ""
        if numRows == 1:return s
        s_Rows = [""] * numRows
        i  = 0
        n = len(s)
        while i < n:
            for j in range(numRows):
                if i < n:
                    s_Rows[j] += s[i]
                    i += 1
            for j in range(numRows-2,0,-1):
                if i < n:
                    s_Rows[j] += s[i]
                    i += 1
        return "".join(s_Rows)