Leetcode刷题之Z字形变换

1、序言

        好几天没有刷题了，最近模型上线有点忙，继续学习。

2、题目

      将一个给定字符串 s 根据给定的行数 numRows，以从上往下、从左到右进行 Z 字形排列。比如输入字符串为"PAYPALISHIRING"，行数为3时，排列如下：

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

之后，你的输出需要从左往右逐行读取，产生出一个新的字符串，比如："PAHNAPLSIIGYIR"。

3、思路

     先设置一个表，用来存储每一个字符元素，设置两个边界条件，一个是达到行数numrows的时候需要换方向倒着走，等倒着走到另外一个边界条件（行为0）的时候又需要正着走，在每一次换方向的时候，列的索引加一。

4、解题

class Solution:
    def convert(self, s: str, numRows: int) -> str:
        if numRows==1:
            return s
        cols = len(s) # 列
        rows = numRows #行
        table = [["" for j in range(cols)] for i in range(rows)]
        row_inx = 0
        col_inx = 0
        flag = 1
        for i, ele in enumerate(s):
            table[row_inx][col_inx]=ele
            row_inx += flag
            if row_inx == rows:
                flag = -flag
                row_inx = row_inx - 2
                col_inx += 1
            if row_inx == 0:
                flag = -flag
                col_inx += 1
        
        result = ''
        for row_list in table:
            row_str = "".join(row_list)
            result += row_str
        return result

5、结果

Accepted
1157/1157 cases passed (688 ms)
Your runtime beats 6.75 % of python3 submissions
Your memory usage beats 5 % of python3 submissions (21.8 MB)