Go實現算法:Z 字形變換(LeetCode)

題目:

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

解題思路:找到規律,每一行添加兩個跨度為 (n-(i+1))*2 和 2*i的字符。

func convert(s string, numRows int) string {

l:=len(s)

if l<=1 || numRows==1{

return s

}

ret:=make([]byte,l)

ret=ret[:0]

for i:=0;i

index:=-1

newIndex:=i

flag:=true

for newIndex

if newIndex>index{

index=newIndex

ret=append(ret,s[index])

}

if flag{

if gap:=(numRows-i-1)*2;gap>0{

newIndex=index+gap

}

flag=false

}else{

if gap:=2*i;gap>0{

newIndex=index+gap

}

flag=true

}

}

}

return string(ret)

}

執行用時: 8 ms, 在所有 golang 提交中擊敗了82.39%的用戶

內存消耗: 3.9 MB, 在所有 golang 提交中擊敗了96.92%的用戶

Go實現算法:Z 字形變換(LeetCode)


分享到:


相關文章: