LeetCode刷题笔记第6题:Z字形变换

LeetCode刷题笔记第6题:Z字形变换

想法:
要完成字符串根据给定的行数从上往下,从左到右完成Z字形排列。当只有一行时直接返回原字符串,当行数大于1时,先以行数构建一个行数数值个空字符串的列表,表示将排列之后的每行字符存入对应位置的列表的位置中。因为在排列的时候,元素的列数在遇到Z字拐角处递增和递减交替进行,用flag来控制列数的递增和递减。详细注释见代码。

class Solution:
    def convert(self, s: str, numRows: int) -> str:
        # 当给定行数为1时,返回原字符串
        if numRows < 2: 
            return s
        # 以列数构建列表中的位置数,为之后存放每行数据做准备
        res = ["" for _ in range(numRows)]
        # 设置初始值,flag用于Z字形排列在拐角处的操作
        i, flag = 0, -1
        for c in s:
            # 将每行的数据加入列表对应的位置处
            res[i] += c
            # 此处判断完成Z字形排列的拐角转行
            if i == 0 or i == numRows - 1: flag = -flag
            i += flag
        # 将每行的字符串(每个位置字符串)拼接
        return "".join(res)

你可能感兴趣的:(LeetCode刷题,leetcode,算法,职场和发展)