方法一:直接构造ans。
发现每行每个字符之间间隔的并不是固定的,所以感觉不太好做,放弃了。。。
方法二:直接模拟。
class Solution {
public:
string convert(string s, int numRows) {
int n = s.size(), index = 0, x = 0, y = 0;
if (numRows == 1) return s; //这里必须特殊判断一下,如果不判断的话进入下面的模拟就出问题了!!!
vector<vector<char>> mp(numRows, vector<char>(n));
while (index < n) {
for (int i = 0; i < numRows && index < n; ++i) {
mp[x][y] = s[index++];
x++;
}
x -= 2;
y++;
for (int i = 0; i < numRows - 2 && index < n; ++i) {
mp[x][y] = s[index++];
x--;
y++;
}
}
string ans;
for (int i = 0; i < numRows; ++i) {
for (int j = 0; j < n; ++j) {
if (mp[i][j] != 0) //0可以写成 '\0' 但是不能写成 ''!!!
//这里必须加一个判断,因为string添加了一个空字符后长度会增加,跟之前不一样了!!!
ans += mp[i][j];
}
}
return ans;
}
};
方法三:空间优化。
具体思路:在方法二中的矩阵有大量的空间没有被使用,能否优化呢?我们可以将矩阵的每行初始化为一个空列表,每次向某一行添加字符时,添加到该行的列表末尾即可。
class Solution {
public:
string convert(string s, int numRows) {
int n = s.size(), index = 0, x = 0, y = 0;
if (numRows == 1) return s;
vector<string> mp(numRows); //换成空string,每次在末尾添加字符即可。
while (index < n) {
for (int i = 0; i < numRows && index < n; ++i) {
mp[x] += s[index++];
x++;
}
x -= 2;
y++;
for (int i = 0; i < numRows - 2 && index < n; ++i) {
mp[x] += s[index++];
x--;
y++;
}
}
string ans;
for (int i = 0; i < numRows; ++i) {
ans += mp[i];
}
return ans;
}
};