【力扣算法09】之 6. N 字形变换 python

**力扣算法09 之6. N 字形变换**

### 题目描述给定一个字符串 `s` 和一个整数 `n`,将字符串 `s` 按照 N 字形进行排列。

例如,当 `s = "PAYPALISHIRING"` 时,如果 `n =3`,则按如下方式排列:

P A H NI T S IY P L R I H G E S

class Solution:
 def convert(self, s: str, numRows: int) -> str:
 # 如果行数为1或2,直接返回原字符串 if numRows ==1 or numRows ==2:
 return s # 初始化一个列表,用于存储每一行的字符 rows = [''] * numRows # 初始化索引和方向 index, step =0,1 # 遍历原字符串中的每个字符 for char in s:
 # 将当前字符添加到对应行中 rows[index] += char # 如果是第一行或最后一行,改变方向 if index ==0:
 step =1 elif index == numRows -1:
 step = -1 # 更新索引 index += step # 将每一行的字符连接起来,得到结果字符串 return ''.join(rows)

solution = Solution()

# 测试用例1s = "PAYPALISHIRING"
numRows =3print(solution.convert(s, numRows)) # 输出: "PAHNAPLSIIGYIR"

# 测试用例2s = "A"
numRows =1print(solution.convert(s, numRows)) # 输出: "A"

# 测试用例3s = "AB"
numRows =1print(solution.convert(s, numRows)) # 输出: "AB"

# 测试用例4s = "ABC"
numRows =2print(solution.convert(s, numRows)) # 输出: "ACB"