力扣<6> Z 字形变换(二维数组)

题目描述:

将一个给定字符串根据给定的行数，以从上往下、从左到右进行 Z 字形排列。
比如输入字符串为 "LEETCODEISHIRING" 行数为 3 时，排列如下：
L C I R
E T O E S I I G
E D H N
之后，你的输出需要从左往右逐行读取，产生出一个新的字符串，比如："LCIRETOESIIGEDHN"。
请你实现这个将字符串进行指定行数变换的函数：
来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/zigzag-conversion
。

解答:

对于我这种脑壳笨的人,看了半天题,一股脑儿只想到了二维数组来解决这个问题,做完之后看别人的,才觉得自己可能做题是做了个寂寞.
二维数组解题思路:

1. 首先定义一个空二维数组,n行(根据题目要求确定),列数我取得题目给的字符串长度,这里坑定浪费空间了,怎么优化目前没想到
2. 双重循环,根据规律可以看出,假设列为i,行为j, 有多少行,则行的倍数列都会放字符,然后i+j等于总行数的位置放字符
3. 正常读取二位数组的字符串拼接即可

def convert(self, s: str, numRows: int) -> str:
        if numRows == 1:
            return s
        n = numRows
        l = len(s)
        s = list(s)
        print(s)
        count = 0
        num_list = [[0 for _ in range(l)] for _ in range(n)]  # 初始化二维数组,行数为年,列数取字符串的长度(基本上浪费很多,怎么优化还没想出来)
        for i in range(l):
            for j in range(n):
                if (i % (n-1)) == 0 and num_list[j][i]==0:  # 行的倍数列
                    num_list[j][i] = s[count]
                    count += 1
                    if count >= l:
                        break
                if (i + j) % (n - 1) == 0 and num_list[j][i]==0:  # 行列角标之和对行总数取余等于0
                    num_list[j][i] = s[count]

                    count += 1
                    if count >= l:
                        break
            if count >= l:
                break
        ans = ''
        for i in num_list:
            for j in i:
                if j != 0:
                    ans += j
        return ans

最后发现结果很感人: image.png

另一种简单解法:

     if numRows == 1:
            return s
        n = numRows
        str_list = ['' for _ in range(n)]
        j = 1
        flag = -1
        for i in s:
            j = j + flag
            str_list[j] += i

            if j >= n-1 or j <= 0:
                flag = -flag
        return ''.join(str_list)