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 numRows) {
int aux = 2*(numRows - 1);
int len = s.size();
string ans = "";
if(aux == 0)
return s;
int m = (len+aux-1)/aux;
for(int i = 0; i < numRows; i++){
for(int j = 0; j < m+1; j++){
if(i == 0){
if(aux*j < len)
ans += s[aux*j];
}
else if(i == numRows-1){
if(aux*j + i < len)
ans += s[aux*j + i];
}
else{
if(j == 0){
if(i < len)
ans += s[i];
}
else{
if(aux*j-i < len)
ans += s[aux*j-i];
if(aux*j+i < len)
ans += s[aux*j+i];
}
}
}
}
return ans;
}
};