6. Z 字形變換

Difficulty: 中等

將一個給定字串根據給定的行數，以從上往下、從左到右進行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

Solution

Language: 全部題目

class Solution:
  def convert(self, s: str, numRows: int) -> str:
    if numRows < 2: return s
    res = ['' for _ in range(numRows)]
    
    i, flag = 0, -1
    for char in s:
      res[i] += char
      if i == 0 or i == numRows - 1: # 如果到了Z字形轉彎的地方，需要反向新增字元
        flag = -flag
      i += flag
    return "".join(res)