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 RAnd 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"
.
我在考虑这个问题的时候,并没有优化。先是列出了一个例子,发现对字符串分组考虑,然后将分组剩下的再额外处理。这样比较易于处理。
先贴出不是很优化的代码吧。
class Solution {
public:
string convert(string s, int nRows) {
if(nRows==1) return s;
vector<string> fraction(nRows,"");
int len=s.length(),index=0;
int fracLen=2*(nRows-1),fracNum=0; //fracLen是分组处理的字符串长度,fracNum是已经处理的组数
while(index<len){
int fracStart=fracNum*fracLen; //这组开始的下标
if((fracStart+fracLen)<len){
for(int i=0;i<(nRows-1);i++){
fraction[i].push_back(s[index++]);
}
for(int i=(nRows-1);i>=1;i--){
fraction[i].push_back(s[index++]);
}
fracNum++;
}
else{ //剩下的元素处理
for(int i=0;i<(nRows-1);i++){
if(index>=len) break;
fraction[i].push_back(s[index++]);
}
for(int i=(nRows-1);i>=1;i--){
if(index>=len) break;
fraction[i].push_back(s[index++]);
}
}
}
string result="";
for(int i=0;i<nRows;i++){
result+=fraction[i];
}
return result;
}
};