算法练习 —— Z-字型变换

Z-字型变换：

此题解法仅为为个人理解，如有其他思路或解法欢迎来探讨！
力扣链接：Z 字形变换

---

题目：

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

比如输入字符串为"PAYPALISHIRING" 行数为3时，排列如下：

P   A   H   N
A P L S I I G
Y   I   R

之后，你的输出需要从左往右逐行读取，产生出一个新的字符串，比如："PAHNAPLSIIGYIR"。

请你实现这个将字符串进行指定行数变换的函数：

String convert(String s, int numRows);

---

示例：

示例 1：

输入：s = "PAYPALISHIRING", numRows = 3
输出："PAHNAPLSIIGYIR"

示例 2：

输入：s = "PAYPALISHIRING", numRows = 4
输出："PINALSIGYAHRPI"
解释：
P     I    N
A   L S  I G
Y A   H R
P     I

示例 3：

输入：s = "A", numRows = 1
输出："A"

---

分析：

由于题目没有要求输出Z-字型变换后的样式，仅要输出最终的字符串结果。所以我做这道题的时候其实是在找Z-字型变换的规律：

我们以题目所给的输入字符串为例，当输入字符串s="PAYPALISHIRING"，numRows = 4时，Z-字型变换得到的结果为"PINALSIGYAHRPI"：

输入：s = "PAYPALISHIRING", numRows = 4
Z-字型变换排列解释：
P →→→ I →→→ N
A → L S → I G
Y A → H R
P →→→ I
输出："PINALSIGYAHRPI"

这里我们不要被Z-字型变换的形状所影响，我们可以让其形状压缩一下，输出还是保持不变的，看起来更方便理解：

P   I   N
A L S I G
Y A H R
P   I
输出："PINALSIGYAHRPI"

再将每一个字符的替换成他所在字符串的下标，得到下面这样排列的一个字串。此时我们可以发现，较长的那几列之间对应位置的差值为6，多看几种排列后总结得到公式为DValue = 2 * numRows - 2(减2是因为第二列的第一行和最后一行没有元素)：

0     6      12
1  5  7  11  13
2  4  8  10
3     9
差值：DValue = 2*4-2

此时，我们将第一列和第二列看作一个整体，再将这组字串分成头、身、尾三部分，并找到其规律，发现每个元素都可以写成i * DValue + j及其的变种；而且在身体部分，我们发现第一列的j1和第二列的j2满足j1 + j2 = DValue：

//最先输出头(i * DValue)
0 * 6 + 0 |           | 1 * 6 + 0 |           | 2 * 6 + 0

//然后输出身（i * DValue+j）
0 * 6 + 1 | 0 * 6 + 5 | 1 * 6 + 1 | 1 * 6 + 5 | 2 * 6 + 1
0 * 6 + 2 | 0 * 6 + 4 | 1 * 6 + 2 | 1 * 6 + 4 | 

//最后输出尾（i * DValue + numRows - 1）
0 * 6 + 3 |           | 1 * 6 + 3 |           |

现在规律已经找到，理清思路，开始写代码

---

代码：

先看上面的分析！自己理解思考一下，不要急着看代码！

public String convert(String s, int numRows) {  //s为输入数组，numRows为要求列数
        //当s的长度小于numRows与numRows==1时，直接返回s
        if (s.length() < numRows || numRows == 1){
            return s;
        }

        //两列编号之差
        int DValue = 2 * numRows - 2;
        //创建一个数组对象用来存放拼接后的数组。对象能够被多次的修改，并且不产生新的未使用对象。
        //由于 StringBuilder 相较于 StringBuffer 有速度优势，所以用StringBuilder
        StringBuilder newStr = new StringBuilder();

        //输出第一行(i*DValue)
        for (int i=0 ; i * DValue < s.length() ; i++){
            //最后一列第一个数字小于数组总长时进入循环
            newStr.append(s.charAt(i * DValue));//charAt是返回指定索引处的字符；append是将读出的数据存入此序列中
        }

        //输出中间（i*DValue+j）
        for (int j=0 ; j < numRows - 2 ; j++){
            for (int i = 0; i * DValue  + j + 1 < s.length(); i++ ){
                //不是第一行也不是最后一行的进入循环
                newStr.append(s.charAt(i * DValue  + j + 1));//只输出1,3,5,……列

                if (i * DValue + (DValue - j - 1) < s.length()){
                    newStr.append(s.charAt(i * DValue + (DValue - j - 1)));
                }
            }
        }

        //输出最后一行（i*DValue+numRows-1）
        for (int i=0 ; i * DValue + numRows - 1 < s.length() ; i++){
            //最后一列最后一个数字小于数组总长时进入循环
            newStr.append(s.charAt(i * DValue + numRows -1));
        }

        return newStr.toString();
    }

最终在力扣得到的运行效率：