Leetcode #6 ZigZag Conversion

题目

The string “PAYPALISHIRING” is written in a zigzag pattern on a given number of rows like this: (you may want to display this pattern in a fixed font for better legibility)

P   A   H   N
A P L S I I G
Y   I   R

And then read line by line: “PAHNAPLSIIGYIR”
Write the code that will take a string and make this conversion given a number of rows:

string convert(string text, int nRows);

convert(“PAYPALISHIRING”, 3) should return “PAHNAPLSIIGYIR”.

思路

观察结果，发现字符串位置序列重复周期为2*nRows-2；
开一个字符串列表，遍历原字符串s，将对应位置的字母添加到列表中对应的字符串中，将列表中字符串连接起来即可；
最后别忘了尾巴。

代码：

class Solution(object):
    def convert(self, s, numRows):
        """
        :type s: str
        :type numRows: int
        :rtype: str
        """
        if numRows == 0:
            return ""
        if numRows == 1:
            return s

        T = 2 * numRows - 2;
        strlist = []
        for i in range(numRows):
            strlist.append("")

        fre = 1
        while((fre * T) < len(s)):
            for i in range(T):
                if i < numRows:
                    strlist[i] += s[(fre - 1)* T + i]
                else:
                    strlist[numRows-2-i%numRows] += s[(fre - 1) * T + i]
            fre += 1

        if len(s) > (fre-1) * T:
            for i in range(len(s) - (fre-1) * T):
                if i < numRows:
                    strlist[i] += s[(fre - 1)* T + i]
                else:
                    strlist[numRows-2-i%numRows] += s[(fre - 1) * T + i]

        result = ""
        for i in range(numRows):
            print(strlist[i])
            result += strlist[i]

        return result