问题原始链接 https://leetcode.com/problems/zigzag-conversion
字符串"PAYPALISHIRING"按给定行数,之字形输出,即先从上往下,再从下往右上,再从上往下...
P A H N A P L S I I G Y I R
然后按行从上往下输出:"PAHNAPLSIIGYIR"。
第一种方法:
按行从输入字符串中查找字符,拼接到结果字符串。
Leet Code 测试耗时 44ms。大量的字符串拼接导致性能低。
public class Solution {
public String convert(String s, int numRows) {
if (s == null || s.length() == 0 || numRows == 1) {
return s;
}
String result = "";
int m = 2 * numRows - 2;
for (int i = 0; i < numRows; i++) {
for (int j = 0;; j++) {
int start = j * m;
if (start >= s.length()) {
break;
}
int k1 = start + i;
if (k1 < s.length()) {
result += s.charAt(k1);
}
if (i != 0 && i != numRows - 1) {
int k2 = start + m - i;
if (k2 != k1 && k2 < s.length()) {
result += s.charAt(k2);
}
}
}
}
return result;
}
}
第二种方法:
使用StringBuilder,按行从输入字符串中查找字符,添加到StringBuilder中,最后一次性构造结果字符串输出。
Leet Code 测试耗时 10ms。
public class Solution {
public String convert(String s, int numRows) {
if (s == null || s.length() == 0 || numRows == 1) {
return s;
}
StringBuilder sb = new StringBuilder();
int m = 2 * numRows - 2;
for (int i = 0; i < numRows; i++) {
for (int j = 0;; j++) {
int start = j * m;
if (start >= s.length()) {
break;
}
int k1 = start + i;
if (k1 < s.length()) {
sb.append(s.charAt(k1));
}
if (i != 0 && i != numRows - 1) {
int k2 = start + m - i;
if (k2 != k1 && k2 < s.length()) {
sb.append(s.charAt(k2));
}
}
}
}
return sb.toString();
}
}
第三种方法:
构造指定行数个StringBuffer,扫描输入字符串,把每个字符添加到对应行的StringBuffer,最后把所有行的StringBuffer拼接成一个字符串。
Leet Code 测试耗时 14ms。
public class Solution {
public String convert(String s, int numRows) {
if (s == null || s.length() == 0 || numRows == 1) {
return s;
}
StringBuilder[] sbs = new StringBuilder[numRows];
for (int i = 0; i < numRows; i++) {
sbs[i] = new StringBuilder();
}
int m = 2 * numRows - 2;
for (int i = 0; i < s.length(); i++) {
int j = i % m;
sbs[j < numRows ? j : m - j].append(s.charAt(i));
}
for (int i = 1; i < numRows; i++) {
sbs[0].append(sbs[i]);
}
return sbs[0].toString();
}
}
第四种方法:
构造一个长度等于输入字符串长度的结果字符数组,构造一个索引数组记录每行下个字符对应的结果字符数组的索引,扫描输入字符串,把字符放入结果字符数组中相应位置。
Leet Code 测试耗时 7ms。
public class Solution {
public String convert(String s, int numRows) {
if (s == null || s.length() == 0 || numRows == 1) {
return s;
}
char[] result = new char[s.length()];
int[] index = new int[numRows];
int m = 2 * numRows - 2;
int times = (s.length() - 1) / m;
int firstRowLastIndexInS = times * m;
int left = s.length() - firstRowLastIndexInS - 1;
for (int i = 1; i < numRows; i++) {
int preRowCount = 0;
if (i - 1 == 0) {
preRowCount = times + 1;
} else {
preRowCount = times == 0 ? 0 : 2 * times;
if (left >= i - 1) {
preRowCount++;
}
if (left >= numRows && i - 1 >= m - left) {
preRowCount++;
}
}
index[i] = index[i - 1] + preRowCount;
}
for (int i = 0; i < s.length(); i++) {
int j = i % m;
int row = j < numRows ? j : m - j;
result[index[row]] = s.charAt(i);
index[row] = index[row] + 1;
}
return new String(result);
}
}