力扣刷题-Z字形字符串变形:按行排序

题目描述

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

比如输入字符串为 “LEETCODEISHIRING” 行数为 3 时,排列如下:

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

实例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

解题思路
1)建立numRows个列表ListA[i],每个列表用来存储相应行的字符串。
2)如果只有一个字符(numRows==1)直接return。
3)设置flag参数,初始为False,作为是否继续按列表的正向存储的标志。
4)遍历字符串,没当index为numRows-1的整数倍的时候,flag反制。之后根据flag的情况按顺序存储字符到相应的列表。
5)将列表组合起来并输出。

T     T     T
T  F  T  F  T
T F   T F   T
F     F     F

可以看到,在flag为True的时候,ListA[i]从0-3行正向存储,在flag为False时,ListA[i]从4-2行反向存储,故,程序可以如下设计:

class Solution:
    def convert(self, s: str, numRows: int) -> str:
        ListA= [[]for i in range(numRows)]  
        if numRows==1:
            return s
        flag= False          
        for i,string in enumerate(s):         
            j = i%(numRows-1)    
            if j == 0:                       
                flag = not flag
            if flag:
                ListA[j].append(string)         
            else:
                ListA[numRows - 1 - j].append(string)
        cache = [''.join(ListA[i]) for i in range(numRows)]
        return ''.join(cache)

参考别人写的,自己重新理解了一遍

你可能感兴趣的:(力扣刷图,python)