LeetCode-in-All

6. Zigzag Conversion

Medium

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:

To solve the Zigzag Conversion problem in Swift using a Solution class, we’ll follow these steps:

  1. Define a Solution class with a method named convert.
  2. Create an array of strings to represent each row of the zigzag pattern.
  3. Initialize variables to keep track of the current row (row) and the direction of traversal (down).
  4. Iterate through each character in the input string s.
    • Append the current character to the string representing the current row.
    • If we reach the first or last row, change the direction of traversal accordingly.
    • Update the current row based on the direction of traversal.
  5. Concatenate the strings representing each row to form the final zigzag conversion.
  6. Return the concatenated string.
  7. Handle edge cases where the number of rows is 1 or the input string is empty.

Here’s the implementation:

class Solution {
    func convert(_ s: String, _ numRows: Int) -> String {
        let arr = Array(s)
        var result = ""
        let shift = numRows > 1 ? (numRows - 1) * 2 : 1
        var index = 0
    
        for row in 0..<numRows { // Step 1
            index = row // Step 2
            
            while index < arr.count {
                result.append(arr[index]) // Step 3
                if row != 0 && row != numRows - 1 { // Step 4
                    let diagonalIndex = index + (numRows - 1 - row) * 2
                    if diagonalIndex < arr.count {
                        result.append(arr[diagonalIndex])
                    }
                }
                index += shift // Step 5
            }
        }
        return result
    }
}

This implementation provides a solution to the Zigzag Conversion problem in Swift.