📑 题目:6. Z 字形变换

🚀 本题 LeetCode 传送门

题目大意

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

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

  1. P A H N
  2. A P L S I I G
  3. Y I R

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

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

  1. string convert(string s, int numRows);

解题思路

  • 这一题没有什么算法思想,考察的是对程序控制的能力。用 2 个变量保存方向,当垂直输出的行数达到了规定的目标行数以后,需要从下往上转折到第一行,循环中控制好方向ji

代码

  1. package leetcode
  2. func convert(s string, numRows int) string {
  3. matrix, down, up := make([][]byte, numRows, numRows), 0, numRows-2
  4. for i := 0; i != len(s); {
  5. if down != numRows {
  6. matrix[down] = append(matrix[down], byte(s[i]))
  7. down++
  8. i++
  9. } else if up > 0 {
  10. matrix[up] = append(matrix[up], byte(s[i]))
  11. up--
  12. i++
  13. } else {
  14. up = numRows - 2
  15. down = 0
  16. }
  17. }
  18. solution := make([]byte, 0, len(s))
  19. for _, row := range matrix {
  20. for _, item := range row {
  21. solution = append(solution, item)
  22. }
  23. }
  24. return string(solution)
  25. }