Leetcode6. Z 字形变换-python

难度：中等

将一个给定字符串根据给定的行数，以从上往下、从左到右进行 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->n->1->n...这么循环的和一群WWWWW似的，而且拼接的时候并没有空格，那么更容易了，不需要考虑空格（输入要构造图像的话明显比输出字符串复杂），我只需要循环字符串的个数，共n行，每次将一个字母连接到对应行即可，以上图为例则效果图如下：

LDR
EOEII
ECIHN
TSG

最后拼接即可，不过关键是如何控制好是WWWW走向～可以用flag=1和-1来代表，每当到第一层的时候就要往下走，每当走到最后一层时就要往上走。

代码如下：

class Solution(object):
    def convert(self, s, numRows):
        """
        :type s: str
        :type numRows: int
        :rtype: str
        """
        #1层的话直接返回即可
        if numRows==1:
            return s
        X=['' for _ in range(numRows)]
        layer=0
        for i in s:
            #第一层就往下走
            if layer==0:
                flag=1
            #最后一层就往上走
            elif layer==numRows-1:
                flag=-1
            X[layer]=X[layer]+i
            layer=layer+flag
        #拼接
        S=''.join(X)
        return S