6.ZigZag Conversion
Description
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 s, int numRows);
Example
Input: s = "PAYPALISHIRING", numRows = 3
Output: "PAHNAPLSIIGYIR"
Input: s = "PAYPALISHIRING", numRows = 4
Output: "PINALSIGYAHRPI"
Explanation:
P I N
A L S I G
Y A H R
P I
Solution: (Java)
class Solution {
public String convert(String s, int numRows) {
if (numRows == 1) {
return s;
}
StringBuilder result = new StringBuilder();
int zrows, gaplength, index=0;
zrows = numRows - 2;
gaplength = numRows + zrows;
for (int i = 0; i < numRows; i++){
if (i != 0 && i != numRows-1) {
index = i;
int count = 0;
int firstadd = gaplength - i*2;
while (index < s.length()) {
result.append(s.charAt(index));
count++;
if (count%2 == 0) {
index += (gaplength-firstadd);
} else {
index += firstadd;
}
}
} else {
index = i;
while (index < s.length()) {
result.append(s.charAt(index));
index += gaplength;
}
}
}
return result.toString();
}
}
思路:按行访问,找出规律即可,特别注意 numRows == 1
这种特殊情况,否则会进入死循环导致内存溢出。