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".

类似Z的倒转，利用down来判断方向。

class Solution {
public:
    string convert(string s, int nRows) {
        int len = s.length();
        if(nRows == 1 || len < 3)//"AB",n
        {
            return s;
        }
        bool down = true;
        vector<vector<char> >table(nRows);
        int row = 0;
        for(int i = 0; i < len;i++)
        {
            if(down && row < nRows)
            {
                table[row++].push_back(s[i]);
            }
            else if(!down && row >= 0)
            {
                table[row--].push_back(s[i]);
            }
            else
            {
                if(down){
                    row -= 2;
                }
                else{
                    row += 2;
                }
                down = !down;
                i--;
            }
        }
        string result = "";
        vector<vector<char> >::iterator it1;
        vector<char>::iterator it2;
        for(it1 = table.begin(); it1 != table.end(); it1++)
        {
            for(it2 = (*it1).begin(); it2 != (*it1).end(); it2++)
            {
                result += *it2;
            }
        }
        return result;
    }
};