Z字形变换

题目描述

解题思路

解析参考自acwing，作者：adnil8130

Python代码

class Solution:
    def convert(self, s: str, numRows: int) -> str:
        if numRows<=0: return ''
        if numRows==1: return s
        r = numRows
        res = ''
        for i in range(0, numRows):
            for j in range(i, len(s), 2*numRows-2):
                if i==0 or i==r-1: 
                    res += s[j]
                else:
                    res += s[j]
                    if j+2*(r-i-1)<len(s):
                        res += s[j+2*(r-i-1)]
        return res

代码描述

第一行和最后一行属于同一类，中间行属于另外一类。在计算中间行的时候，先添加当前竖杠位置的字符，然后判断右侧相邻的斜杠位置是否存在。这样从上到下按行（0，r），从左到右按列（i，len(s)），不会遗漏也不会越界。