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) {
if (nRows == 1) return s;
vector<string> vec(nRows);
for (int i = 0, j = 0, k = 1; s[i]; ++i, j += k) {
vec[j] += s[i];
if (j == nRows - 1)
k = -1;
else if (j == 0)
k = 1;
}
string ans;
for (int i = 0; i < nRows; ++i)
ans += vec[i];
return ans;
}
};