将一个给定字符串 s 根据给定的行数 numRows ,以从上往下、从左到右进行 Z 字形排列
思路:依次遍历字符串中的每个字符,根据此时所处的行来决定下一次的前进方向,具体代码如下:
class Solution {
public:
string convert(string s, int numRows) {
int sz = s.size();
if (numRows == 1) {
return s;
}
vector<string> vec(numRows);
bool goDown = false;
int row = 0;
for (int i = 0; i < sz; ++i) {
vec[row] += s[i];
if (row == 0 || row == numRows - 1) {
goDown = !goDown;
}
row += goDown ? 1 : -1;
}
string res;
for (auto& str : vec) {
res += str;
}
return res;
}
};