leetcode必回:N 字形变换

200 阅读1分钟

开启掘金成长之旅!这是我参与「掘金日新计划 · 2 月更文挑战」的第 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"

 

提示:

  • 1 <= s.length <= 1000
  • s 由英文字母(小写和大写)、',' 和 '.' 组成
  • 1 <= numRows <= 1000

分析

根据题目的分析,我们如何求n字形的变化?

看了题目其他什么想法都没有,唯一想法就是开辟足够大的二维数组,按照题目意思存入数组,最后按要求输出。 根据题目给定的numRows开辟j = numRows行, 根据题目意思:最开始正序正常存入,但当j == numRows时,开始逆序存入,最后按行输出,搞定。

解答

   class Solution(object):
    def convert(self, s, numRows):
        """
        :type s: str
        :type numRows: int
        :rtype: str
        """
        if numRows == 1 or len(s) <= numRows:return s

        #开辟足够大的二维空间
        c = [[] for _ in range(numRows)]
        j = 0
        #设置陷阱 flag值为0时正序 值为1时逆序
        flag = 0
        for i in range(len(s)):
            c[j].append(s[i])
            #控制移动方向
            if flag == 0:
                j+=1
            elif flag == 1:
                j-=1
            #选择排序方向
            if j % numRows == 0 and j != 0 and numRows>2:
                j-=2
                flag = 1
            elif j % numRows ==0 and j!= 0 and numRows == 2:
                j-=2
            elif j == 0:
                flag = 0
        #list[list] 转变为 list
        res = [''.join(i) for i in c]
        #list 转变为str
        ans = "".join(res).strip("[").strip("]")
        return ans

总结

上面的算法思路让我学到了许多知识啊