【Leetcode每日一题】mid——6. N 字形变换

题目描述

https://leetcode.cn/problems/zigzag-conversion/description/
将一个给定字符串 s 根据给定的行数 numRows ，以从上往下、从左到右进行 Z 字形排列。

比如输入字符串为 “PAYPALISHIRING” 行数为 3 时，排列如下：

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

请你实现这个将字符串进行指定行数变换的函数：

string convert(string s, int numRows);

示例 1：

输入：s = “PAYPALISHIRING”, numRows = 3
输出：“PAHNAPLSIIGYIR”

Tag: 字符串处理

class Solution:
    def convert(self, s: str, numRows: int) -> str:
        if numRows <= 1:
            return s
        
        l1 = numRows
        l2 = numRows-2
        flag = 0
        slice1 = []
        slice2 = []
        length = len(s)
        while flag < length:
            if flag + l1 < length:
                slice1.append(s[flag:flag + l1])
                flag += l1
            else:
                slice1.append(s[flag:length])
                break
            if flag + l2 < length:
                slice2.append(s[flag:flag + l2])
                flag += l2
            else:
                slice2.append(s[flag:length])
                break
        flag1=len(slice1)
        flag2=len(slice2)
        flag = max(flag1, flag2)
        result = ""
        for i in range(numRows):
            for j in range(flag):
                if j<flag1:
                    if i < len(slice1[j]):
                        result += slice1[j][i]
                if i > 0 and i < numRows-1:
                    if j<flag2:
                        if len(slice2[j])- i >= 0:
                            result += slice2[j][len(slice2[j])-i]
        return result

思路分析

• 构造场景模拟处理方法
• 寻找数学规律，推导下标变换公式

实际上我给出的方法类似于两种结合，但是很遗憾没能通过测试（解答错误 539 / 1157 个通过的测试用例）
通过画图+模拟，可以把这个过程拆分成：

1. 新建两个字符串队列（一列的I 和斜着的 /)
2. 根据不同的下标从队列中按照题目要求顺序进行字符提取
   但是这样做既没能利用程序的自动化，也没整明白数学公式，本来想结合两种方法的长处，现在变成了结合两种方法的短处了。经过多次调试也没弄明白，证明此路不通。

所以要么构造矩阵，要么把数学规律整明白。

14点12分模拟法

看官方版本最直接

V字形变换法

参考这位的思路，把找循环规律做到了极致

class Solution:
    def convert(self, s: str, numRows: int) -> str:
        if numRows <= 1:
            return s
        length = 2*(numRows-1)
        str_length = len(s)
        loop = math.ceil(str_length/length)
        result = ""
        for j in range(0, numRows):
            for i in range(0, loop):
                tmp1 = i*length + j
                tmp2 = i*length + (length-j)
                # print(f"{length} {tmp1} {tmp2}")
                if j == 0 or j == numRows-1:
                    if tmp1 < str_length:
                        result += s[tmp1]
                else:
                    if tmp1 < str_length:
                        result += s[tmp1]
                    if tmp2 < str_length:
                        result += s[tmp2]
        return result