ZigZag Conversion

題目意思是將字符串變成Z型字符串

這道題一開始我是想着通過計算下一步取的字符所在的位置來構成zigzag字符串。

解題思路:
1. 首先進行特殊情況處理,如果len(字符串長度)小於行數,或者行數等於1輸出它本身
2. 接下來分成s.length() / numRows個小片,每片長度爲2*(numRows-1)
3. 由於在切片過程中容易忽略掉s.length()==sliceLen + 1的情況,以及在行數等於2的時候第一行需要取多一位數,因此在這裏設置flag變量。
4. 接下來就設置兩次循環,外循環控制行數,內循環控制slice的index
5. 返回string ans

後來看了一下討論區的思路,又要被自己蠢哭了…其實完全可以模擬zigzag的書寫過程來存儲string,根據討論區的思路,自己又重寫了一遍:

class Solution {
public:
    string convert(string s, int numRows) {
        // good solution in discussion
        if (numRows <= 1)
            return s;
        string *buf = new string[numRows];
        string ans = "";
        int step = 0; // using step to control the zigzag order
        int row = 0;
        for (int i = 0; i < s.length(); i++) {
            buf[row] += s[i];
            if (row == 0)
                step = 1;
            else if (row == numRows - 1)
                step = -1;
            row += step;
        }

        for (int i = 0; i < numRows; i++) 
            ans += buf[i];

        return ans;
        /* solved by myself
        if (s.length() < numRows || numRows == 1)
            return s;
        string ans = "";
        int sliceLen = 2 * (numRows - 1);
        int flag = (s.length() == sliceLen + 1 || numRows == 2)? 1: 0;
        for (int i = 0; i < numRows; i++) {
            for (int j = 0; j < (s.length() + flag) / numRows; j++) {
                if (!i && sliceLen * j < s.length()) 
                    ans += s[sliceLen * j];
                else {
                    if (sliceLen * j + i < s.length())
                        ans += s[sliceLen * j + i];
                    if (sliceLen * (j + 1) - i < s.length() && sliceLen != 2 * i)
                        ans += s[sliceLen * (j + 1) - i];
                }
            }
        }
        return ans;
        */
    }
};
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章