leetcode No6. ZigZag Conversion

Question：

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"

.

即把字符串N字型排列，然后一行一行的输出。

Algorithm：

暴力法，其实可以发现这些排列的规律，利用数学知识求解，但是写这个排列还是费了我一些功夫。

用一个二维数组存储N字排列，一共有nRows行，（s.size()/2）+1列。

然后再遍历二维数组。

Submitted Code：

class Solution {   //暴力法   速度太慢
public:
string convert(string s, int numRows) {
if(numRows==1)return s;
vector<vector<char>> array(numRows);
string t;
for(int i=0;i<numRows;i++)
array[i].resize((s.size()/2)+1);
int flag=0;  //转折点，下或上
int x=0,y=0; //行，列
for(int j=0;j<s.size();j++)
{
if (x>(numRows-1))
{
flag = 1;
x = numRows-2;
y++;
}
if (x<0)
{
flag = 0;
x = 1;
y--;
}
array[x][y]=s[j];
if(flag==0)
x++;
else
{
x--;
y++;
}
}
for(int p=0;p<numRows;p++)
for(int q=0;q<((s.size()/2)+1);q++)
{
if(array[p][q]>='!'&&array[p][q]<='~')
t.push_back(array[p][q]);
}
return t;
}
};