6. ZigZag Conversion Leetcode Python

最新推荐文章于 2019-11-09 17:42:37 发布

hyperbolechi

最新推荐文章于 2019-11-09 17:42:37 发布

阅读量1.3k

点赞数

分类专栏： leetcode 文章标签： python leetcode

本文链接：https://blog.csdn.net/hyperbolechi/article/details/43217949

版权

leetcode 专栏收录该内容

166 篇文章 0 订阅

订阅专栏

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 "PAHNAPL

SIIGYIR".

这题的主要特点是要观察出新序列的位置为 2*nRows-2

代码如下

class Solution:
    # @return a string
    def convert(self, s, nRows):
        if nRows==1:
            return s
        if len(s)<=1:
            return s
        n=nRows
        size=2*n-2
        result=[]
        for i in range(n):
            j=i
            while j<len(s):
                result.append(s[j])
                if i!=0 and i!=n-1 and j+size-2*i<len(s):
                    result.append(s[j+size-2*i])
                j+=size
        return "".join(result)