mirror of
https://github.com/SunnyQjm/algorithm-review.git
synced 2026-06-03 08:16:43 +08:00
86 lines
2.5 KiB
Python
86 lines
2.5 KiB
Python
#!/usr/bin/env python
|
||
# coding=utf-8
|
||
|
||
############################################################
|
||
# Z 字形变换
|
||
# 将一个给定字符串根据给定的行数，以从上往下、从左到右进行 Z 字形排列。
|
||
# 比如输入字符串为 "LEETCODEISHIRING" 行数为 3 时，排列如下：
|
||
# L C I R
|
||
# E T O E S I I G
|
||
# E D H N
|
||
# 之后，你的输出需要从左往右逐行读取，产生出一个新的字符串，比如："LCIRETOESIIGEDHN"。
|
||
# 请你实现这个将字符串进行指定行数变换的函数：
|
||
# string convert(string s, int numRows);
|
||
#
|
||
# 示例 1：
|
||
# 输入: s = "LEETCODEISHIRING", numRows = 3
|
||
# 输出: "LCIRETOESIIGEDHN"
|
||
#
|
||
# 示例 2:
|
||
# 输入: s = "LEETCODEISHIRING", numRows = 4
|
||
# 输出: "LDREOEIIECIHNTSG"
|
||
# 解释:
|
||
#
|
||
# L D R
|
||
# E O E I I
|
||
# E C I H N
|
||
# T S G
|
||
############################################################
|
||
|
||
import collections
|
||
|
||
|
||
class Solution:
|
||
def convert(self, s, numRows):
|
||
"""
|
||
:type s: str
|
||
:type numRows: int
|
||
:rtype: str
|
||
|
||
(Knowledge)
|
||
|
||
算法思路:
|
||
|
||
1. 用numRows个list记录每行的字符串；
|
||
|
||
2. 然后顺序遍历字符串，按Z字形顺序放到合适的行中；
|
||
|
||
3. 最后将每行的字符串拼接即可
|
||
"""
|
||
if numRows == 1:
|
||
return s
|
||
|
||
lines = [[] for i in range(numRows)]
|
||
|
||
# 当前的方向（首先向下，触底向上，触顶向下，依次改变方向）
|
||
goDown = True
|
||
|
||
# 下一个字符要写入的行号，初始为0,根据方向进行更新，向下则+1,向上则-1
|
||
lineNumber = 0
|
||
|
||
# 依次遍历字符串，按Z字型顺序写到合适的行
|
||
for i in range(len(s)):
|
||
lines[lineNumber].append(s[i])
|
||
|
||
# 判断是否需要改变方向
|
||
if goDown and lineNumber == numRows - 1:
|
||
goDown = False
|
||
if not goDown and lineNumber == 0:
|
||
goDown = True
|
||
|
||
# 根据当前的方向更新行号
|
||
lineNumber = lineNumber + 1 if goDown else lineNumber - 1
|
||
|
||
# 将所有numRows个list拼接成一个list
|
||
for i in range(1, numRows):
|
||
lines[0] += lines[i]
|
||
|
||
# 将list转字符串返回
|
||
return "".join(lines[0])
|
||
|
||
|
||
if __name__ == '__main__':
|
||
solution = Solution()
|
||
print(solution.convert("LEETCODEISHIRING", 3), "= \nLCIRETOESIIGEDHN")
|
||
print(solution.convert("LEETCODEISHIRING", 4), "= \nLDREOEIIECIHNTSG")
|