Z 字形变换

Z 字形变换

将一个给定字符串根据给定的行数，以从上往下、从左到右进行 Z 字形排列。

输入: s = “LEETCODEISHIRING”, numRows = 4
输出: “LDREOEIIECIHNTSG”

L     D     R
E   O E   I I
E C   I H   N
T     S     G

代码+注释思路:

class Solution {
    public String convert(String s, int numRows) {
         //算出所有列的列数，竖列元素个数为n，斜列元素个数为1,每一个竖列后跟着n-2个斜列
        int len = s.length();
        int columnNum = 0;
        int shuColumnNum = numRows;
        int slash = numRows - 2;
        while (len > 0) {
            len = len - shuColumnNum;
            columnNum++;
            for (int i = slash; i > 0 && len > 0; i--) {
                len--;
                columnNum++;
            }
        }
        //有了列数和行数，总个数=列*行
        int total = numRows * columnNum;
        char[] result = new char[total];
      
        //开始插入字符
        //插入的当前列
        int curColumn = 0;
        int index = 0;
        char[] chars = s.toCharArray();
        while (index < chars.length) {
            //填竖列，每个竖列是三个数，分别计算3个数的索引，填入
            for (int i = 0; i < numRows && index < chars.length; i++) {
                result[columnNum * i + curColumn] = chars[index++];
            }
            //竖列填完了，开始往后curColumn++，n-2个斜列，每个斜列1个
            curColumn++;
            //每循环一次填一个斜列，curColumn++
            for (int i = slash; i > 0 && index < chars.length; i--) {
                result[columnNum * i + curColumn] = chars[index++];
                curColumn++;
            }
        }

       StringBuilder sb=new StringBuilder();
        for (char c:result)
        {
            if (c!='\u0000')
            {
                sb.append(c);
            }
        }
        return sb.toString();
    }
}

最大循环执行的次数就是行数列数，n代表字符串中字符的个数
行数为row，列数大致等于n/(2row-2)(row-1)即n/2，总次数就是n*row/2，当n足够大时忽略系数，时间复杂度为O(n)