Leetcode 6.Z字形变换题解
题目链接:
https://leetcode-cn.com/problems/zigzag-conversion/
题目描述
将一个给定字符串根据给定的行数,以从上往下、从左到右进行 Z 字形排列。
比如输入字符串为 “LEETCODEISHIRING” 行数为 3 时,排列如下:
之后,你的输出需要从左往右逐行读取,产生出一个新的字符串,比如:“LCIRETOESIIGEDHN”。
思路
这道题很明显是一道找规律的题目,首先看样例输出的第一行,每个字符在原字符串中的位置分别是0,4,8,12,每个字符间都差4位即2*(行数-1)。然后再往下看,虽然第二行的字符多了很多,但是e,o,s,i这四个字符之间依然只是差了四位,只不过每个字符间又多了一个字符,而多的字符的位置也与当前行数是相关的,很容易看出其规律。因此按行遍历z形字符串即可得到问题解。
代码
class Solution(object):
def convert(self, s, numRows):
"""
:type s: str
:type numRows: int
:rtype: str
"""
if numRows == 1:
return s
res = ''
#按行遍历
for i in range(numRows):
now = i
while now < len(s):
res += s[now]
tmp = 2*(numRows-1)-2*i
if 0<i<numRows-1 and now+tmp < len(s):
res += s[now+tmp]
now += 2*(numRows - 1)
return res