LeetCode 6. ZigZag Conversion(Python)兼翻译

6. ZigZag Conversion

转化成锯齿形

本题来自LeetCode OJ

题目翻译

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)
给定行数,将字符串"PAYPALISHIRING" 写成锯齿的形式,如下图所示:

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

And then read line by line: "PAHNAPLSIIGYIR"
然后一行一行的读取则为:
"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".

注:ZigZag是锯齿形,也就是每条边是相同的字符个数

题目分析

根据锯齿状的结构,我们以字符串长度为10,ZZ n行为例:
第一行:第一个下标索引为0,下一个为竖下来这笔需要n-1个,斜过去这笔为n-1,所以第二个下标索引为2(n-1).我们很简单就可以看出规律,是(n-1)的倍数。
所以第二行呢?
第二行的情况一共有两种:
– 竖下来的第一行的下一个
– 斜上去第一行的上一个
也就是说是(n-1)整除余1或者余(n-2)
……
所以规律出来了:

if (i % nextnum) == j or (i % nextnum) == nextnum-j:
                        ZigZag += s[i]
                        dellist.append(i)

但是一个i在某一行就必定不在别的行,所以接下来我们进行删除操作,这样越到后面代码所需要迭代的次数就越少了。

代码示例

class Solution(object):
    def convert(self, s, numRows):
        """ :type s: str :type numRows: int :rtype: str """
        if len(s) <= numRows or numRows == 1:
            ZigZag = s
        else:
            ZigZag = ""
            Candidate = [i for i in range(len(s))]
            nextnum = 2*(numRows-1)
            for j in range(numRows):
                dellist = []
                for i in Candidate:
                    if (i % nextnum) == j or (i % nextnum) == nextnum-j:
                        ZigZag += s[i]
                        dellist.append(i)
                for i in dellist:
                    Candidate.remove(i)
        return ZigZag

自己心累

最开始写的时候用错了remove,倒是程序总是得不到我想要的答案。不能够直接remove,我猜应该是remove了之后,for i in Candidate中的索引就变了。

点赞