Leetcode 6. ZigZag Conversion

class Solution {
public:
string convert(string s, int numRows) {
if(numRows==1) return s;
string *a=new string[numRows];
int b=numRows*2-2;
for(int i=0;i<s.size();i++)
{
int j=i%b;
if(j<numRows)
a[j].push_back(s[i]);
else
a[b-j].push_back(s[i]);
}
string s0;
for(int i=0;i<numRows;i++)
s0+=a[i];
return s0;
}
};