Create 0006-zigzag-conversion.go

Author: Youssef Eddaif

Diff for: go/0006-zigzag-conversion.go

@@ -0,0 +1,25 @@
+package zigzagconversion
+
+import (
+	"strings"
+)
+
+func convert(s string, numRows int) string {
+	if numRows == 1 {
+		return s
+	}
+	var result strings.Builder
+	listLength := len(s)
+	step := (numRows - 1) * 2
+	for i := 0; i < numRows; i++ {
+		// reset steps at last row
+		for k := i; k < listLength; k += step {
+			result.WriteByte(s[k])
+			if i < numRows-1 && i > 0 && k+step-i*2 < listLength {
+				result.WriteByte(s[step+k-i*2])
+			}
+
+		}
+	}
+	return result.String()
+}