6.Z 字形变换

将一个给定字符串 s 根据给定的行数 numRows ，以从上往下、从左到右进行 Z 字形排列。

比如输入字符串为 "PAYPALISHIRING" 行数为 3 时，排列如下：

P A H N A P L S I I G Y I R

之后，你的输出需要从左往右逐行读取，产生出一个新的字符串，比如："PAHNAPLSIIGYIR"。

请你实现这个将字符串进行指定行数变换的函数：

string convert(string s, int numRows);

示例 1：

输入：s = "PAYPALISHIRING", numRows = 3 输出："PAHNAPLSIIGYIR"

示例 2：

输入：s = "PAYPALISHIRING", numRows = 4 输出："PINALSIGYAHRPI" 解释： P I N A L S I G Y A H R P I

示例 3：

输入：s = "A", numRows = 1 输出："A"

提示：

1 <= s.length <= 1000 s 由英文字母（小写和大写）、',' 和 '.' 组成 1 <= numRows <= 1000

解题思路：

  • 观察规律: 仔细观察 Z 字形排列规律，你会发现字符串在 Z 字形路径上移动，每次移动都是先向下，再斜向上。
  • 模拟路径: 我们可以使用 numRows 个字符串数组来存储每一行的字符。遍历字符串 s，根据当前行数和移动方向，将字符添加到对应行的数组中。
  • 拼接结果: 最后，将所有行的字符串数组依次拼接起来，即可得到最终的 Z 字形变换后的字符串。
func convert(s string, numRows int) string {
	  // 长度为1和numRows小于原字符串长度，无需变换
	  if numRows == 1 || len(s) <= numRows {
	  	return s
	  }
      
	  // 定义 numRows 长度一维数组用于保存变换的字符子串
	  rows := make([]string, numRows)
	  // 一维数组元素位置
	  var index = 0
	  // step 用于区别遍历的方向
	  var step = 1
	  for i := 0; i < len(s); i++ {
	  	rows[index] += string(s[i])
      
	  	// 如果到达数组首位，也改变遍历方向，反方向
	  	if index == 0 {
	  		step = 1
	  	}
	  	// 如果到达数组末位，改变遍历方向
	  	if index == numRows-1 {
	  		step = -1
	  	}
      
	  	index += step
	  }
      
	  var ans string
	  for _, row := range rows {
	  	ans += row
	  }
	  return ans
}