題目意思是將字符串變成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;
*/
}
};