题目描述
将一个给定字符串根据给定的行数，以从上往下、从左到右进行 Z 字形排列。
比如输入字符串为 “LEETCODEISHIRING” 行数为 3 时，排列如下：
L C I R
E T O E S I I G
E D H N
之后，你的输出需要从左往右逐行读取，产生出一个新的字符串，比如：”LCIRETOESIIGEDHN”。
请你实现这个将字符串进行指定行数变换的函数：
string convert(string s, int numRows);
示例
示例 1:
输入: s = "LEETCODEISHIRING", numRows = 3
输出: "LCIRETOESIIGEDHN"
示例 2:
输入: s = "LEETCODEISHIRING", numRows = 4
输出: "LDREOEIIECIHNTSG"
解释:
L D R
E O E I I
E C I H N
T S G
|
解法
func convert(s string, numRows int) string {
var (
convertStr = ""
convertSubStrList = make([]string, numRows)
)
if numRows < 1 {
return ""
} else if numRows == 1 {
return s
}
for index := 0; index < len(s); index++ {
k := index % (2*numRows - 2)
y := 2*numRows - 2 - k
if k >= 0 && k < numRows {
convertSubStrList[k] += string(s[index])
} else if y >= 0 && y < numRows {
convertSubStrList[y] += string(s[index])
}
}
convertStr = strings.Join(convertSubStrList, "")
return convertStr
}
|
本地测试
package main
import (
"fmt"
"strings"
)
func main() {
s, l := "LEETCODEISHIRING", 3
fmt.Println(convert(s, l), "LCIRETOESIIGEDHN")
s, l = "LEETCODEISHIRING", 4
fmt.Println(convert(s, l), "LDREOEIIECIHNTSG")
}
|
来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/zigzag-conversion
