将一个给定字符串s根据给定的行数numRows,以从上往下、从左到右进行Z字形排列。
比如输入字符串为 “PAYPALISHIRING” 行数为 3 时,排列如下:
之后,你的输出需要从左往右逐行读取,产生出一个新的字符串,比如:“PAHNAPLSIIGYIR”。
思路一:
从左到右按箭头方向迭代s,将每个字符添加到合适的行。之后从上到下遍历行即可。
我们假定n = numRows
class Solution {
public:
string convert(string s, int numRows) {
if (numRows == 1) return s;
vector<string> rows(min(numRows, int(s.size()))); // 防止s的长度小于行数
int curRow = 0;
bool goingDown = false;
for (char c : s) {
rows[curRow] += c;
if (curRow == 0 || curRow == numRows - 1) {// 当前行curRow为0或numRows -1时,箭头发生反向转折
goingDown = !goingDown;
}
curRow += goingDown ? 1 : -1;
}
string ret;
for (string row : rows) {// 从上到下遍历行
ret += row;
}
return ret;
}
};
因为只需遍历一次,所以时间复杂度为O(len(s))。
思路二:
假定有numRows = 3,那么
第0行:0 - 4 - 8 - 12 - (下标间距都是4)
第1行:1 - 3 - 5 - 7 - (下标间距都是2)
第2行:2 - 6 - 10 - 14 -(下标间距都是4)
这还看不出什么规律来,假定numRows = 4,那么
第0行:0 - 6 - 12 - 18 - (下标间距都是6)
第1行:1 - 5 - 7 - 11 - 13(下标间距有变化:4 2 4 2 ……)
第2行:2 - 6 - 10 - 14 -(下标间距都是4)
可以得到以下结论:
- 起始下标都是行号
- 第0层和第numRows - 1层的下标间距总是step。
- 中间层的下标间距总是step - 2 * 行数,2 * 行数交替
- 下标不能超过len(s) - 1
string convert(string s, int numRows) {
if (numRows == 1) return s;
int step = numRows * 2 - 2; // 间距
int index = 0;// 记录s的下标
int len = s.length();
int add = 0; // 真实的间距
string ret;
for (int i = 0; i < numRows; i++) // i表示行号
{
index = i;
add = i * 2;
while (index < len)//超出字符串长度计算下一层
{
ret += s[index]; // 当前行的第一个字母
add = step - add;// 第一次间距是step -2*i,第二次是2*i,
index += (i == 0 || i == numRows-1) ? step : add; // 0行和最后一行使用step间距,其余使用add间距
}
}
return ret;
}