The string “PAYPALISHIRING” is written in a zigzag pattern on a given number of rows like this: (you may want to display this pattern in a fixed font for better legibility)
P A H N
A P L S I I G
Y I R
And then read line by line: “PAHNAPLSIIGYIR”
Write the code that will take a string and make this conversion given a number of rows:
string convert(string text, int nRows);
convert(“PAYPALISHIRING”, 3) should return “PAHNAPLSIIGYIR”.
题目举的例子不够明显,我们用数字,做一个4行的图形更容易分析。
0 | - | - | 6 | - | - | 12 |
1 | - | 5 | 7 | - | 11 | 13 |
2 | 4 | - | 8 | 10 | - | 14 |
3 | - | - | 9 | - | - | 15 |
首先我们观察到间隔着有些列是完整的,我们按照完整行对字符串进行分段组,每组具有numRows+numRows-2个元素。
继续观察,每一个分组中,第一行和最后一行都只有一个元素,而其余各行都有两个元素,两个元素的间隔offset = 2*numRows-row-row-2.于是我们得到如下的代码:
class Solution {
public:
string convert(string s, int numRows){
size_t len = s.size();
string result;
if(len <= numRows|| numRows ==1)
return s;
int groupLen = 2*numRows -2;
for(int row=0; row<numRows; row++){
if (row==0 || row== numRows-1) {
for (int i=row; i<len; i+= groupLen) {
result.push_back(s[i]);
}
}else{
int offset = 2*numRows -2 -2 *row;
for (int i=row; i<len; i+=groupLen) {
result.push_back(s[i]);
if(i+offset <len)
result.push_back(s[i+offset]);
}
}
}
return result;
}
};
上面这段代码虽然能够被AC,但是,但中间那个if-else显得很丑陋,其实row=0时,offset=2*numRows-2,就是下一次分割的第一个元素。row=numRows-1时,offset=0,就是本身。这样可以将那个if-else合并起来,不过仍然注意避免越界。得到下面的代码,这个看起来要美观一点:
class Solution {
public:
string convert(string s, int numRows){
size_t len = s.size();
string result;
if(len <= numRows|| numRows ==1)
return s;
int groupLen = 2*numRows -2;
for (int row=0; row<numRows; row++) {
int offset = 2*numRows -2 -2*row;
for(int i=row; i<len; i+=groupLen){
result.push_back(s[i]);
if(offset!=0 && offset!=groupLen && i+offset<len){
result.push_back(s[i+offset]);
}
}
}
return result;
}
};