蓝桥杯训练--力扣6

#题目

6. N 字形变换

将一个给定字符串 s 根据给定的行数 numRows ，以从上往下、从左到右进行 Z 字形排列。

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

P A H N

A P L S I I G

Y I R

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

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

string convert(string s, int numRows);

示例 1：

输入：s = "PAYPALISHIRING", numRows = 3

输出："PAHNAPLSIIGYIR"

示例 2：

输入：s = "PAYPALISHIRING", numRows = 4

输出："PINALSIGYAHRPI"

解释：

P I N

A L S I G

Y A H R

P I

示例 3：

输入：s = "A", numRows = 1

输出："A"

#思想

找到这些相邻元素的关系

1. 观察可得第一行和最后一行的排列方式相同，但和中间排列方式不同

2. 先找第一行和最后一行的相邻关系，发现俩个相邻的元素间隔2*numRows - 2个位置

3. 再找中间元素关系，第一个元素到第二个元素是从下往上走，第二个元素到第三个元素是从上往下走，呈现交替的关系，就可以设置一个标志，像开关一样

#代码（带详细注释）

class Solution:
    def convert(self, s: str, numRows: int) -> str:
        ans = [] #最后输出的
        if numRows != 1:
            for i in range(numRows): #循环输出每一层,让层数和加一方便计算
                if i == 0 or i == numRows - 1: #第一层和最后一层的排列方式不一样
                    # for j in range(len(s) // numRows): #循环列
                    for k in range(i,len(s),2*numRows -2): #找到第一排和最后一排规律，最后一排是从i开始
                        ans.append(s[k]) #添加到
                else: #中间层数
                    d = i #中间元素下标
                    flag = 0 #开始是从下到上

                    while True:

                        if d >= len(s):
                            break #大于s就下一行
                        if flag == 0: #斜线的点按从上到下还是从下到上
                            ans.append(s[d])
                            d = d + 2*(numRows-i-1)#相邻元素的规律，每循环一次就加一个
                            flag = 1
                        elif flag == 1: #开始从上到下
                            ans.append(s[d])
                            d = d + 2*i
                            flag = 0
        
            return ''.join(ans)
        else:
            return s

#运行结果