【leetcode】6 Python实现Z字型变换

题目描述

将字符串 “PAYPALISHIRING” 以Z字形排列成给定的行数：

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

之后从左往右，逐行读取字符：“PAHNAPLSIIGYIR”

实现一个将字符串进行指定行数变换的函数:

string convert(string s, int numRows);

示例 1:

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

示例 2:

输入: s = “PAYPALISHIRING”, numRows = 4
输出: “PINALSIGYAHRPI”
解释:
P I N
A L S I G
Y A H R
P I

思路

我们考查示例2，从第零列开始，将偶数列和奇数列放在一块，可以知道，每一块有b=（4*2-2）个字符 ,(其中4为numRows的值），因为字符串中有自己的索引值i，因此对于偶数列的字符，我们可以通过i%b的值来定位到。比如对于第零行中的偶数列的字符，其索引值均满足i%b== 0, 第一行的偶数列的字符，其索引值均满足i%b== 1，以此类推，一直到i%b== 4-1,(其中4为numRows的值)。对于奇数列中的字符，我们不这样定位，因为要保证在偶数列中的字符被定位到时，同行的奇数列也要同时被定位到。因此可以通过偶数列的字符来定位奇数列的字符，例如偶数列字符此时索引值为i，则与它同行的奇数列的索引值则为（i+(numRows-1-i)*2） (可以自己找个例子试下）。

于是，我们可以创建一个空数组，在满足i%b==0到numRows-1 时，把字符加进去，其中在第一行到第（numRows-1）行时，把偶数列中的字符加进列表时，同时用上面的方法，把奇数列中的字符也加进去。注意，最后的偶数列和奇数列块时，用（i+(numRows-1-i)*2）索引时需要判断一下是否超出给定字符串的范围。

代码

class Solution:
    def convert(self, s, numRows):
        n=len(s)
        if numRows==1:
            return s
        else:
            a=[]
            b=2*numRows-2 #算出除数
            for j in range(n):
                if j%b==0:
                    a.append(s[j])
            for i in range(1,numRows-1):
                for j in range(n):
                    if j%b==i:
                        a.append(s[j])
                        if 2*(numRows-i-1)+j<n:
                            a.append(s[2*(numRows-i-1)+j])
            for j in range(n):
                if j%b==(numRows-1):
                    a.append(s[j])
            q=''.join(a)       #把列表变成字符串   
            return q