leetcode day 13【 6. Z 字形变换】flag的巧妙运用及多种情况的巧妙合并

最新推荐文章于 2024-11-01 15:19:21 发布

皮皮鲁与鲁西西�

最新推荐文章于 2024-11-01 15:19:21 发布

阅读量829

点赞数

分类专栏：每天都刷leetcode 文章标签： leetcode 算法职场和发展

本文链接：https://blog.csdn.net/weixin_44398263/article/details/121430874

版权

每天都刷leetcode 专栏收录该内容

28 篇文章 0 订阅

订阅专栏

在这里插入图片描述

解题思路

以新字符串为基础，找到新字符串的逐个元素的在原字符串中的位置，逐一添加字符到新字符串中
除了首行和尾行的offset是2n-2外，其余行都是两个offset（2n-2i-2，2i）轮流来
虽然实现了，但是总感觉flag那里写的十分幼稚和啰嗦
复杂度分析：原字符串中每个字符被访问一次，因此时间o(n)，空间o(1)

class Solution:
    def convert(self, s: str, numRows: int) -> str:
        if numRows == 1: return s
        newstring=''
        for row in range(numRows):
            i = row
            if row == 0 or row == numRows-1:
                offset = 2*numRows - 2
                while i < len(s):
                    newstring += s[i]
                    # newstring.join(s[i])
                    i += offset
            else:
                offset1 = 2*numRows -2*row - 2
                offset2 = 2*row
                flag = 0
                while i < len(s):
                    newstring += s[i]
                    if not flag:
                        i += offset1    
                        flag = 1
                    else:
                        i += offset2
                        flag = 0
        return newstring

后来发现offset1+offset2=2n-2，所以可以巧妙利用一下;中间行和首尾行的情况也可以合并一下：

class Solution:
    def convert(self, s: str, numRows: int) -> str:
        if numRows == 1: return s
        newstring=''
        for row in range(numRows):
            i = row
            offset = 2*row
            while i < len(s):
                newstring += s[i]
                offset = 2*numRows-2  if row == 0 or row == numRows-1 else 2*numRows-2-offset
                i += offset
        return newstring

以原字符串为基础，逐一对字符串分配到不同行的列表（字符串）中。最后将各行字符串拼接。
其中flag用得十分巧妙，无需对行增和行减两种情况分开讨论。
复杂度分析：原字符串中每个字符被访问分配一次，因此时间o(n)，额外使用numRows个记录每行元素，因此空间o(n)

class Solution:
    def convert(self, s: str, numRows: int) -> str:
        if numRows == 1: return s
        newstringlist = ["" for _ in range(numRows)]
        row, flag = 0, -1
        for i in range(len(s)):
            newstringlist[row] += s[i]
            if row == 0 or row == numRows - 1:
                flag *= -1
            row += flag
        return "".join(newstringlist)