leetcode ZigZag Conversion

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"

.
我在考虑这个问题的时候，并没有优化。先是列出了一个例子，发现对字符串分组考虑，然后将分组剩下的再额外处理。这样比较易于处理。

先贴出不是很优化的代码吧。

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;
}
};