首先还是暴力,先构建一个 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 浪费时间的优化,有空在看吧,佛系刷题