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%的用户。

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

猜你喜欢

转载自www.cnblogs.com/qinvis/p/11974291.html