【LeetCode】6. N 字形变换

1 问题

将一个给定字符串 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”

2 答案

我写的不对

class Solution:
    def convert(self, s: str, numRows: int) -> str:
        l = len(str)
        a_z = 2*numRows-2  # 一个z字母所需长度
        numbers_of_z = l//a_z  # z的数量
        mod_of_z = l%a_z  # 会剩下几个组成不了z的字母
        conlumns_z = numRows -1 # 一个z所占列数
        list = [[None * (conlumns_z+mod_of_z) for i in range(numRows)]]  # 生成适合大小的列表
        k = 0
        for j in range(conlumns_z*numbers_of_z):
            for i in range(numRows):

                if j % (numRows-1) == 0:
                    list[i][j] = s[k]
                elif j % (numRows-1) == 1: 

            k+=1
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17

利用flag，其实很简单

class Solution:
    def convert(self, s: str, numRows: int) -> str:
        if numRows < 2: return s
        res = ["" for _ in range(numRows)]
        i, flag = 0, -1  # i是列表的索引
        for c in s:
            res[i] += c
            if i == 0 or i == numRows - 1: flag = -flag
            i += flag
        return "".join(res)
1
2
3
4
5
6
7
8
9
10

3 总结

看到题目先观察思考，巧解