leetcode第6题：Z字形变换--直接模拟求解法

【题目描述】

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

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

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

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

string convert(string s, int numRows);

【解题思路】

对于Z字形，每行使用一个StringBuiler来记录，所以得到一个数组StringBuiler[numRows]；通过数组下标y，记录当前方向的up，开始模拟构造Z字型。

【代码】

String convert(String s, int numRows) {

    if (numRows == 1) {

        return s;

    }

    StringBuilder[] builders = new StringBuilder[numRows];

    for (int i = 0; i < numRows; i++) {

        builders[i] = new StringBuilder();

    }

    // 核心代码快

    char[] arr = s.toCharArray();

    int y = 0;

    boolean up = false;

    for (int i = 0; i < arr.length; i++) {

        builders[y].append(arr[i]);

        if (up) { // 自下向上

            y--;

            if (y == -1) {

                y = 1;

                up = false;

            }

        } else { // 自上向下

            y++;

            if (y == numRows) {

                y = numRows - 2;

                up = true;

            }

        }

    }

    StringBuilder result = new StringBuilder(numRows);

    for (StringBuilder b: builders) {

        result.append(b.toString());

    }

    return result.toString();

}

【运行结果】

执行用时 : 6 ms, 在所有 java 提交中击败了85.36%的用户；

内存消耗 : 39 MB, 在所有 java 提交中击败了93.94%的用户。

【小结】

算法的执行效率有待提高，但较易于理解。大家如有好的看法，欢迎补充。