LeetCode ZigZag Conversion

LeetCode解题之ZigZag Conversion

原题

将所给字符串按特定形状排列，依次将每排的字符连接形成新的字符串。具体看一个例子：给定字符串”PAYPALISHIRING”，相应排列成的形状如下：

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

可以看出来是将字符串按照N的书写顺序重新排列了。现在要求我们从第一排开始依次把所有字符组成新的字符串：”PAHNAPLSIIGYIR”

注意点：

• 行数不是固定的

例子：

输入: s=”PAYPALISHIRING”, numRows=3
输出: “PAHNAPLSIIGYIR”

解题思路

发现其实新的字符串的顺序在原字符串中是有规律可循的。重新组合后的形状其实就是一个个N，相邻N对应位置之间的字符个数是固定的，如例子中的P-A-H-N之间的字符都是3个。现在假设行数为n，则对应位置字符间的间距都为2n-3，现在第一排和最后一排的字符顺序就可以确定了；但中间排的字符就比较特殊，在对应位置之间还会夹着一个字符。以例子中第一列的A和第二列的P来说，假设A的下标为i（从0开始），则A和P的间距为2(n-i-1)-1=2n-2i-3，而A和L的间距为2n-3，同时去掉P自己，所以P和L的间距为(2n-3)-(2n-2i-3)-1=2i-1。这样只需遍历第一列的字符，并依次加上对应的间距来获取后面的字符即可。上文的间距是指中间隔了几个字符。

AC源码

class Solution(object):
    def convert(self, s, numRows):
        """ :type s: str :type numRows: int :rtype: str """
        if numRows<=1:
            return s
        result = ''
        index = 0
        n = len(s)
        for i in range(0, numRows):
            if i == 0 or i == numRows - 1:
                while index < n:
                    result += s[index]
                    index += 2 * numRows - 2
                index = i + 1
            else:
                while index < n:
                    result += s[index]
                    index += 2 * numRows - 2 * i - 2
                    if index >= n:
                        break
                    result += s[index]
                    index += 2 * i
                index = i + 1
        return result

# Test cases
if __name__ == "__main__":
    Solution().convert("PAYPALISHIRING", 2) == "PYAIHRNAPLSIIG"
    Solution().convert("PAYPALISHIRING", 3) == "PAHNAPLSIIGYIR"
    Solution().convert("PAYPALISHIRING", 4) == "PINALSIGYAHRPI"

欢迎查看我的Github (https://github.com/gavinfish/LeetCode-Python) 来获得相关源码。