python leetcode 6. Z 字形变换

最新推荐文章于 2024-11-11 08:42:13 发布

原创最新推荐文章于 2024-11-11 08:42:13 发布 · 129 阅读

0 ·

CC 4.0 BY-SA版权

将一个给定字符串根据给定的行数，以从上往下、从左到右进行 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

在这里插入图片描述

按照图1的推导，就可以逐行获取新字符串对应原字符串的下标
ROW=1是一个特例,单独处理：

if numRows == 1 : return s

class Solution:
    def convert(self, s: str, numRows: int) :
        if numRows == 1 : return s
        strLen = len(s)
        res = ''
        for row in range(1, numRows + 1):
            col = 0
            while True:
                topCol = (2 * numRows - 2) * col + 1
                if row !=1 and row !=numRows and col !=0:
                    tmpIndex = topCol-row   # (topCol - row + 1) - 1
                    if tmpIndex < len(s):
                        res += s[tmpIndex]
                    else:
                        break
                tmpIndex = topCol+row -2    # (topCol+row -1) - 1
                if tmpIndex < strLen:
                    res += s[tmpIndex]
                else:
                    break
                col += 1
        return res