leetcode第6題:Z字形變換--直接模擬求解法

【題目描述】
將一個給定字符串根據給定的行數,以從上往下、從左到右進行 Z 字形排列。
好比輸入字符串爲 "LEETCODEISHIRING" 行數爲 3 時,排列以下:
java

以後,你的輸出須要從左往右逐行讀取,產生出一個新的字符串,好比:"LCIRETOESIIGEDHN"。
請你實現這個將字符串進行指定行數變換的函數:
string convert(string s, int numRows);算法

【解題思路】
對於Z字形,每行使用一個StringBuiler來記錄,因此獲得一個數組StringBuiler[numRows];經過數組下標y,記錄當前方向的up,開始模擬構造Z字型。數組

【代碼】app

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%的用戶。函數

【小結】
算法的執行效率有待提升,但較易於理解。你們若有好的見解,歡迎補充。ui

相關文章
相關標籤/搜索