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

string convert(string s, int nRows) {
if(s.size() <= 1 || nRows <= 1) return s;
int gap = 2*nRows-2;
string res = "";
for(int i=0; i<nRows; i++){
for(int j=i; j<s.size(); j+=gap){
res += s[j];
//画图,e.g: nRows = 3 or 4
if(i>0 && i<nRows-1 && j+gap-2*i < s.size()){
res += s[j+gap-2*i];
}
}
}
return res;
}