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 text, int nRows);
convert("PAYPALISHIRING", 3) should return "PAHNAPLSIIGYIR".
这道题目就是一个找规律的题目,没有其他的方法,题目已经给出nRows = 3的情况,我们可以再写出nRows = 4时的情况。代码如下:
上面的方法,每一行都要遍历一遍字符串中的字符,我们通过观察可以发现,第一行和最后一行的下一个元素都是加上一个next,中间的行在每一个next前面多了一个元素,每次加入next之前单独处理就可以了,这样只需要遍历一遍。代码如下:
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 text, int nRows);
convert("PAYPALISHIRING", 3) should return "PAHNAPLSIIGYIR".
这道题目就是一个找规律的题目,没有其他的方法,题目已经给出nRows = 3的情况,我们可以再写出nRows = 4时的情况。代码如下:
public class Solution {
public String convert(String s, int numRows) {
if(numRows == 1) return s;
StringBuilder sb = new StringBuilder();
int next = 2 * numRows - 2;
int mod = 0;
while(mod < numRows) {
for(int i = 0; i < s.length(); i++) {
if(i % next == mod || i % next == next - mod) {
sb.append(s.charAt(i));
}
}
mod ++;
}
return sb.toString();
}
}
上面的方法,每一行都要遍历一遍字符串中的字符,我们通过观察可以发现,第一行和最后一行的下一个元素都是加上一个next,中间的行在每一个next前面多了一个元素,每次加入next之前单独处理就可以了,这样只需要遍历一遍。代码如下:
public class Solution {
public String convert(String s, int numRows) {
if(s == null || s.length() == 0) return "";
if(numRows == 1) return s;
int next = 2 * numRows - 2;
char[] c = new char[s.length()];
int index = 0;
for(int i = 0; i < numRows; i++) {
for(int j = i; j < s.length(); j += next) {
c[index++] = s.charAt(j);
if(i != 0 && i != numRows - 1 && j + next - i * 2 < s.length())
c[index++] = s.charAt(j + next - i * 2);
}
}
return new String(c);
}
}