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 RAnd 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"
.
class Solution {
public:
string convert(string s, int nRows) {
int len = s.size();
int small = max(0, nRows-2);
stringstream oss;
for(int i = 1; i <= nRows; ++i)
for(int j = 1; j <= (len/(small+nRows) + 1)*2; ++j)
if(isChar(i, j, nRows, small))
{
int idx = xy2Int(i, j, nRows, small);
if(idx <= len) oss << s[idx-1];
}
return oss.str();
}
bool isChar(int x, int y, int big, int small)
{
if(1 == y%2)
return x >= 1 && x <= big;
else
return x >= 2 && x <= small + 1;
}
int xy2Int(int x, int y, int big, int small)
{
int idx = 0;
idx += (big + small)*((y-1)/2);
idx += ((y+1)%2)*big;
if(0 == y%2 && x >= 1 && x <= small+1)
idx += small - x + 2;
else
idx += x;
return idx;
}
};