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:
/*algorithm
in normal string s[0...n], assume ith char(i < rows)
the next element in zigzag patten string is
current i
next i + [n+n-2] = i + [2n-2] (i == 0 || i==n-1)
else i + [n-1-i%n + n-1 - i]=i + [n-1-i]*2
eg,n = 3
P A H N
A P L S I I G
Y I R
*/
string convert(string s, int numRows) {
if(numRows < 2)return s;
string ret;
int r = numRows;
int len;
for(int i = 0;i < row;i++){
if(i >= s.size())break;
ret.append(1,s[i]);
int m = (row-1-i)*2;
int n = 2*row-2;
int l2 = 0,l1 = l2;
do{
l1 += m;
l2 += n;
if(0 < i && i < n-1){
if((i+l1) >= s.size())break;
if(l1 != (l2-n))ret.append(1,s[i + l1]);
}
if((i+l2) >= s.size())break;
ret.append(1,s[i + l2]);
l1 = l2;
}while(1);
}
return ret;
}
};