Leetcode6. 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

解题思路

1. 对字符串进行分组
2. 根据行数计算出每一组字符串数
3. 同样根据行数和字符串数计算出Z字形变换的列数
4. 新建一个字符串，迭代遍历Z字形的每一个元素，判断是否存在字符，如果存在计算出该位置原来的字符串的索引，追加到新的字符串。
   值得注意的是Swift字符串访问元素效率很低，可以转换成Character数组，效率是原来的3倍。

Swift代码

class Solution {
        func convert(_ s: String, _ numRows: Int) -> String {
            if s.isEmpty || numRows <= 1 {
                return s
            }
            var array = s.map({$0})
            let groupCount = 2 * numRows - 2
            let columnPerGroup = numRows - 1
            var numColumns = columnPerGroup * (array.count / groupCount)
            let remainder = array.count % groupCount
            numColumns += (remainder / numRows) + (remainder % numRows)
            var string = ""
            for y in 0..<numRows {
                for x in 0..<numColumns {
                    var i = 0
                    let groupIndex = x / columnPerGroup
                    let groupRemainder = x % columnPerGroup
                    if groupRemainder == 0 {
                        i =  groupCount * groupIndex + y
                        if i < array.count {
                            string.append(array[i])
                        }
                    } else {
                        if groupRemainder + y + 1 == numRows {
                            i = groupCount * groupIndex + (groupCount - y)
                            if i < array.count {
                                string.append(array[i])
                            }
                        }
                    }
                    
                }
            }
            return string
        }
    }