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