N字形变换 Golang leecode_6

首先还是暴力，先构建一个 numRows 行，1000列的二维数组，然后将字符串 s 变成字符数组按照 N 字形排列，然后再 Z 字形遍历，代码逐过程调试很好看懂。

package main

import (
	"fmt"
)

func convert(s string, numRows int) string {
	if numRows == 1 || numRows >= len(s) {
		return s
	}

	flag := 0  // 用来控制内嵌 for 循环的扫描方向
	count := 0 // 用作 bytes 字符数组的索引
	j := 0
	dp := make([][]byte, numRows)
	for i := range dp {
		dp[i] = make([]byte, 1000) // 调整大小以避免索引越界
	}
	bytes := []byte(s)

	for i := 0; i < 1000; i++ {
		for ; j < numRows && flag == 0; j++ {
			if count < len(bytes) {
				dp[j][i] = bytes[count]
				count++
			} else {
				break
			}
		}
		flag = 1
		i++
		for j = numRows - 2; j >= 0 && flag == 1; j-- {
			if count < len(bytes) {
				dp[j][i] = bytes[count]
				count++
				i++
			} else {
				break
			}
		}
		flag = 0
		i = i - 2
		j = 1
		if count >= len(bytes) {
			break
		}
	}
	// 构建结果字符串
	var res []byte
	for n := 0; n < numRows; n++ {
		for m := 0; m < 1000; m++ {
			if dp[n][m] != 0 {
				res = append(res, dp[n][m])
			}
		}
	}
	return string(res)
}

func main() {
	s := "PAYPALISHIRING"
	numRows := 3
	fmt.Println(convert(s, numRows))
}

用时 56ms(超过6.12%用户)，内存消耗 6.83MB(超过49.92%的用户)。果然暴力不是最好的方法，看看有没有方法可以进行优化。看到了一个人的题解，思路和我的一样，但是对字符串和二维数组的操作比我牛逼的多，原谅我是golang 小白，下面是代码，很值得我学习

func convert(s string, numRows int) string {
    n, r := len(s), numRows
    if r == 1 || r >= n {
        return s
    }
    t := r*2 - 2
    c := (n + t - 1) / t * (r - 1)
    mat := make([][]byte, r)
    for i := range mat {
        mat[i] = make([]byte, c)
    }
    x, y := 0, 0
    for i, ch := range s {
        mat[x][y] = byte(ch)
        if i%t < r-1 {
            x++ // 向下移动
        } else {
            x--
            y++ // 向右上移动
        }
    }
    ans := make([]byte, 0, n)
    for _, row := range mat {
        for _, ch := range row {
            if ch > 0 {
                ans = append(ans, ch)
            }
        }
    }
    return string(ans)
}

作者：力扣官方题解
链接：https://leetcode.cn/problems/zigzag-conversion/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。

该作者还有一个第二版的优化方案，是针对遍历大多数 0 浪费时间的优化，有空在看吧，佛系刷题