LeetCode 6. Z 字形变换 -Go 实现

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"

func convert(s string, numRows int) string {
    // 很巧妙 头尾转变方向
    arr := make([]string, numRows)
    if numRows < 2 {
        return s
    }

    i, flag := 0, -1
    for _, v := range s {
        arr[i] += string(v)
        if i == 0 || i == numRows -1 {
            flag = -flag
        }
        i += flag
    }

    var result string
    fmt.Println(arr, result)

    for i := 0; i < len(arr); i++ {
        result += arr[i]
    }

    return result
}

参考资料

https://leetcode-cn.com/problems/zigzag-conversion/

https://mp.weixin.qq.com/s/xpS7iLMf6iGIb6fAfLaGqg