题目意思是将字符串变成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;
*/
}
};