题目描述
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 sl1 = numRowsl2 = numRows-2flag = 0slice1 = []slice2 = []length = len(s)while flag < length:if flag + l1 < length:slice1.append(s[flag:flag + l1])flag += l1else:slice1.append(s[flag:length])breakif flag + l2 < length:slice2.append(s[flag:flag + l2])flag += l2else:slice2.append(s[flag:length])breakflag1=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 个通过的测试用例)
通过画图+模拟,可以把这个过程拆分成:
- 新建两个字符串队列(一列的I 和斜着的 /)
- 根据不同的下标从队列中按照题目要求顺序进行字符提取
但是这样做既没能利用程序的自动化,也没整明白数学公式,本来想结合两种方法的长处,现在变成了结合两种方法的短处了。经过多次调试也没弄明白,证明此路不通。
所以要么构造矩阵,要么把数学规律整明白。
14点12分模拟法
看官方版本最直接
V字形变换法
参考这位的思路,把找循环规律做到了极致
class Solution:def convert(self, s: str, numRows: int) -> str:if numRows <= 1:return slength = 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 + jtmp2 = 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