ZigZag Conversion

The string "PAYPALISHIRING" is written in azigzag pattern on a given number of rows like this: (you may want to displaythis pattern in a fixed font for better legibility)

P   A  H   N
A P L S I I G
Y   I  R

And thenread line by line: "PAHNAPLSIIGYIR"

Write the code that will take a string and make this conversiongiven a number of rows:

stringconvert(string text, int nRows);

convert("PAYPALISHIRING",3) should return "PAHNAPLSIIGYIR".

class Solution {
public:
    string convert(string s, int nRows) {
        if (nRows == 1) {
            return s;
        }
        string *arr = new string[nRows];
        for (int i=0; i<nRows; ++i){
            arr[i]= "";
        }
        int len = s.length();
        int step[2]={1,-1};
        for (int i=0,j=0; i<len; ++i) {
            int stepPos = i/(nRows-1)%2;
            arr[j]+=s[i];
            j+=step[stepPos];
        }
        s="";
        for (int i=0; i<nRows; ++i){
            s+=arr[i];
        }
        delete []arr;
        return s;
    }

};

 

And thenread line by line: "PAHNAPLSIIGYIR"

Write the code that will take a string and make this conversiongiven a number of rows:

stringconvert(string text, int nRows);

convert("PAYPALISHIRING",3) should return "PAHNAPLSIIGYIR".