首先还是暴力,先构建一个 numRows 行,1000列的二维数组,然后将字符串 s 变成字符数组按照 N 字形排列,然后再 Z 字形遍历,代码逐过程调试很好看懂。
package mainimport ("fmt"
)func convert(s string, numRows int) string {if numRows == 1 || numRows >= len(s) {return s}flag := 0 // 用来控制内嵌 for 循环的扫描方向count := 0 // 用作 bytes 字符数组的索引j := 0dp := 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 = 1i++for j = numRows - 2; j >= 0 && flag == 1; j-- {if count < len(bytes) {dp[j][i] = bytes[count]count++i++} else {break}}flag = 0i = i - 2j = 1if count >= len(bytes) {break}}// 构建结果字符串var res []bytefor 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 := 3fmt.Println(convert(s, numRows))
}
用时 56ms(超过6.12%用户),内存消耗 6.83MB(超过49.92%的用户)。果然暴力不是最好的方法,看看有没有方法可以进行优化。看到了一个人的题解,思路和我的一样,但是对字符串和二维数组的操作比我牛逼的多,原谅我是golang 小白,下面是代码,很值得我学习
func convert(s string, numRows int) string {n, r := len(s), numRowsif r == 1 || r >= n {return s}t := r*2 - 2c := (n + t - 1) / t * (r - 1)mat := make([][]byte, r)for i := range mat {mat[i] = make([]byte, c)}x, y := 0, 0for 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 浪费时间的优化,有空在看吧,佛系刷题