[golang] 6. Zigzag Conversion

// [url]https://leetcode.com/studyplan/top-interview-150/[/url]
 
package topInterview
 
// convert
//
// 6. Zigzag Conversion
//
// The string "PAYPALISHIRING" is written in a zigzag pattern on a given number of rows like this: (you may want to display this pattern in a fixed font for better legibility)
//
// P   A   H   N
// A P L S I I G
// Y   I   R
// And then read line by line: "PAHNAPLSIIGYIR"
//
// Write the code that will take a string and make this conversion given a number of rows:
//
// string convert(string s, int numRows);
//
// Example 1:
//
// Input: s = "PAYPALISHIRING", numRows = 3
// Output: "PAHNAPLSIIGYIR"
// Example 2:
//
// Input: s = "PAYPALISHIRING", numRows = 4
// Output: "PINALSIGYAHRPI"
// Explanation:
// P     I    N
// A   L S  I G
// Y A   H R
// P     I
// Example 3:
//
// Input: s = "A", numRows = 1
// Output: "A"
//
// Constraints:
//
// 1 <= s.length <= 1000
// s consists of English letters (lower-case and upper-case), ',' and '.'.
// 1 <= numRows <= 1000//
func convert(s string, numRows int) string {
 
    // Если строка одна возвращаем ее сразу
    if numRows == 1 {
        return s
    }
 
    // Создаем слайс строк длиной numRows
    rows := make([]string, numRows)
    // Указательна текущую строку
    currentRowIndex := 0
    // Флаг движения вниз
    directionDown := true
 
    // Цикл по строке
    for i := range len(s) {
 
        // Если указатель строк на первой строке
        if currentRowIndex == 0 {
            // Подымаем флаг движения вниз
            directionDown = true
        }
 
        // Если указатель строк на крайней строке
        if currentRowIndex == len(rows)-1 {
            // Опускаем флаг движения вниз
            directionDown = false
        }
 
        // Добавляем символ в строку на которой указатель текущей строки
        rows[currentRowIndex] += string(s[i])
 
        // Если поднят флаг движения вниз
        if directionDown {
            // Увеличиваем указатель текущей строки
            currentRowIndex++
        } else { // Если не поднят флаг движения вниз
            // Уменьшаем указатель текущей строки
            currentRowIndex--
        }
    }
    
    // Возвращаем строку из объединенного слайса строк  
    return strings.Join(rows, "")
}

func TestConvert(t *testing.T) {
    data := []struct {
        s        string
        rowNums  int
        expected string
    }{
        {"PAYPALISHIRING", 3, "PAHNAPLSIIGYIR"},
        {"PAYPALISHIRING", 4, "PINALSIGYAHRPI"},
        {"A", 1, "A"},
        {"AB", 1, "AB"},
    }
 
    for i, datum := range data {
        result := convert(datum.s, datum.rowNums)
 
        if result != datum.expected {
            t.Errorf(
                "unexpected result for test data index %d (%s), expected [%s], but got [%s]",
                i,
                datum.s,
                datum.expected,
                result,
            )
        }
    }
}