python算法题之《Z字形变换》

题目要求

将一个给定字符串根据给定的行数，以从上往下、从左到右进行 Z 字形排列。

比如输入字符串为 "LEETCODEISHIRING" 行数为 3 时，排列如下：

L   C   I   R
E T O E S I I G
E   D   H   N
之后，你的输出需要从左往右逐行读取，产生出一个新的字符串，比如："LCIRETOESIIGEDHN"。

请你实现这个将字符串进行指定行数变换的函数：

string convert(string s, int numRows);
示例 1:

输入: s = "LEETCODEISHIRING", numRows = 3
输出: "LCIRETOESIIGEDHN"
示例 2:

输入: s = "LEETCODEISHIRING", numRows = 4
输出: "LDREOEIIECIHNTSG"
解释:

L     D     R
E   O E   I I
E C   I H   N
T     S     G

代码及分析

class Solution:
    def convert(self, s, numRows):
        # 返回字符串
        ans = ''
        # 如果行数不等于1
        if numRows!=1:
            x = 2*(numRows-1)
        # 等于1
        else:
            x = 1
        # 一个空的列表，用于顺序存放每一行的字母
        ansL = []
        # 创建numRows字符串
        for i in range(numRows):
            # 默认为空
            ansL.append('')
        # 遍历每个字母的下标
        for i in range(len(s)):
            # 让0 ≤ j < x
            j = i % x
            # 如果j > numRows-1，则将该下标对应的字母放入第x-j行
            if j > numRows-1 :
                ansL[x-j]+=s[i]
            # 反之，放入第j行
            else:
                ansL[j] += s[i]
        # 将ansL中的字符串顺序连接返回
        for i in ansL:
            ans+=i
        return ans

image (1).png

结果

image (2).png