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"

.

用简单的模拟来做，下面附上我的代码：

<pre name="code" class="cpp">class Solution {
public:
char **arr;
string convert(string s, int numRows) {
if (numRows==1||s.length()==0) {
return s;
}
//计算行数列数
int Y;
if ((numRows*2-2)!=0) {
Y=(int)((numRows*s.length())/(numRows*2-2))+1;
}else{
Y=1;
}
int X=numRows;
arr= new char *[X];
for (int i=0; i<X; i++) {
arr[i]=new char[Y];
}
//初始化数组
for (int i=0; i<X; i++) {
for (int j=0; j<Y; j++) {
arr[i][j]=0;
}
}
string ans;
int vertical=1;
int i=0;
int j=0;//数组下标

for (int k =0; k<s.length(); k++) {
arr[i][j]=s[k];
//逐个字符遍历
if (vertical&&i!=numRows-1) {//如果是竖直的并且还没到num个
i++;//下一次摆放的位置
continue;
}
if (vertical&&i==numRows-1) {//如果是竖着的已经num个了

vertical=0;
i--;
j++;//下一次摆放的位置
continue;
}
if (!vertical&&i!=0) {//如果是对角线，没到顶。
i--;j++;//下一次摆放的位置。
continue;
}
if (!vertical&&i==0) {//如果是对角线，到顶了。
i++;//下一次开始一个新的竖着的
vertical=1;
}
}
ans=toString(s.length(),X, Y);
return ans;
}
string toString(int size,int X,int Y){
char temparr[1000];
int count=0;
memset(temparr, 0, sizeof(temparr));
for (int i=0; i<X; i++) {
for (int j=0; j<Y; j++) {
char c=arr[i][j];
if (c!=0) {
temparr[count++]=c;
size--;
if (size==0) {
break;
}
}
}
}
return (string)temparr;
}

};

同学的代码：

其思路是每个循环赋值好一个竖排加上一个对角线：

string convert(string s, int numRows){
if(numRows == 1)
return s;
long len = s.size();
long n = (len/(2*numRows-2) +1)*numRows;
char **array;
array = new char*[numRows];
for(int i = 0; i < numRows; i++){
array[i] = new char
;
}
for(int i = 0; i < numRows; i++)
for(int j = 0; j < n; j++)
array[i][j] = ' ';
int cur = 0, i = 0, j = 0;
while(cur < s.size()){
for(; i < numRows && cur < s.size(); i++, cur++)
array[i][j] = s[cur];
i -= 2;
j++;
for(; i >= 0 && cur < s.size(); i--, j++, cur++)
array[i][j] = s[cur];
i += 2;
}

string s1;
for(i = 0; i < numRows; i++)
for(j = 0; j < n; j++)
if(array[i][j] != ' ')
s1 += array[i][j];
return s1;
}