Leetcode-6. Z字形变换

题目来源：https://leetcode-cn.com/problems/zigzag-conversion/

将一个给定字符串根据给定的行数，以从上往下、从左到右进行 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);

思路：

设res存储Z字形变换后每行的字符，以leetcode为例，设numRows=3：

res[0]：l          c                     输出lc

res[1]：e    t    o    e               输出etoe

res[2]：e         d                     输出ed

可以看到行索引i的变化顺序为：0~n  → n~0 → 0~n，如此循环。

• 那么从第0行开始，每存一个字符，行索引i+1
• 到达最后一行时，反转递增方向，行索引i-1
• 回到第0行时，又反转递增方向，行索引i+1
• .......

设置一个标志位，标志行索引的递增方向，每次触碰边界时反转即可。

代码：

class Solution(object):
    def convert(self, s, numRows):
        """
        :type s: str
        :type numRows: int
        :rtype: str
        """
        if numRows < 2:
            return s
        res = ["" for _ in range(numRows)]
        flag = -1
        i = 0
        for c in s:
            res[i] += c
            if i == 0 or i == numRows - 1:          # 边界，反转标志
                flag *= -1
            i += flag
        return ''.join(res)

参考：

https://leetcode-cn.com/problems/zigzag-conversion/solution/zzi-xing-bian-huan-by-jyd/