6. Z 字形變換

將一個給定字串根據給定的行數，以從上往下、從左到右進行 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

解法:

思路: 等差數列, 控制迴圈變數達到預期效果, 畫兩張圖就能弄出來

參考網址:https://blog.csdn.net/qq_33022911/article/details/83616833

func convert(s string, numRows int) string {

	if numRows <= 1 || len(s) == 0 {
		return s
	}

	rst := ""

	size := 2*numRows - 2

	for i := 0; i < numRows; i++ {
		for j := i; j < len(s); j += size {
			rst += string(s[j])
			tmp := j + size - 2*i
			if i != 0 && i != numRows-1 && tmp < len(s) {
				rst += string(s[tmp])
			}
		}
	}
	return rst
}

大神解法中 8ms 完成用例沒有看懂, 好像是用指標完成的; 貼出來12ms的:

這裡大神用了一個特性: go語言中的string中, 每個字元是可以用 uint8 來直接表示出來的, 這樣就省去了切單個字元 s[j] 時候的強制轉換

執行結果:

LDREOEIIECIHNTSG
uint8

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

	var cs []uint8
	var count int
	var index int
	for i := 0; i < numRows; i++{
		count = 0
		index = count*(2*numRows-2) + i
		for index < len(s){
			cs = append(cs, s[index])
			if i != 0 && i != numRows - 1 {
				index = (count + 1) * (2 * numRows - 2) - i
				if index >= len(s) {
					break
				}
				cs = append(cs, s[index])
			}
			count++
			index = count*(2*numRows-2) + i
		}
	}

	return string(cs)
}